混泥土教学(模拟)
作者:互联网
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);
}
标签:max,ll,个数,腰长,len,教学,泥土,模拟,define 来源: https://www.cnblogs.com/Wang-Xianyi/p/16536304.html