本题主要靠模拟思维……
#include<bits/stdc++.h>
using namespace std;
const int N = 2e6 + 10, M = 998244353;
long long ans, a[N];
int maxn = -1, sum, k = 1;
int main() {
int n; cin >> n;
for(int i = 1; i <= n; i ++ ) {
int b; cin >> b;
a[b] ++;
//最长的一条边
maxn = max(maxn, b);
}
//边的长度
for(int i = 1; i <= maxn; i ++) {
//重新计算可调用的边数(第三边,之前满足<2*i的边也可以用,k不用再从1开始枚举)
for(; (k < 2 * i) && (k <= maxn); k ++) sum += a[k];
//组合:第一种是等腰三角形,第二种 是全等三角形
ans = ans + ((a[i] * (a[i] - 1) * (sum - a[i]) / 2) + (a[i] * (a[i] - 1) * (a[i] - 2) / 6)) % M;
}
cout << ans % M;
}
2021-06-10第一次
2021-06-23第二次