混泥土教学（模拟）

https://www.luogu.com.cn/problem/P6686

• 记录每个长度的棍子的个数
• 枚举每个长度，作为腰长
• 再遍历到腰长两倍以内的，这些累加进去得到满足条件的底的个数
• 第三步不需要从头遍历，累计的次数k也不需要归0，因为小的腰长对应的底边的个数一定在大的腰长对应底边个数以内，就不需要重复计算了
• 再用组合乘法原理求当前腰长对应的贡献量加进去，（分不等边和等边考虑）

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define MAX 10000001
ll datas[MAX];
ll max_len;
ll n, k;
ll ans;
ll t;
ll other = 1;
#define MOD 998244353
inline void input()
{
    scanf("%lld", &n);
    for (ll i = 1; i <= n; i++)
    {
        scanf("%lld", &t);
        datas[t]++;
        if (t > max_len)
        {
            max_len = t;
        }
    }
}
int main()
{
    input();
    for (ll i = 1; i <= max_len; i++)
    {
        for (; other < i * 2 && other <= max_len; other++)
        {
            k += datas[other];
        }
        ans += datas[i] * (datas[i] - 1) * (k - datas[i]) / 2 + datas[i] * (datas[i] - 1) * (datas[i] - 2) / 6;
    }
    printf("%lld", ans % MOD);
}