比赛 NOIP2025模拟赛3 评测结果 AAAAAAAAAA
题目名称 Good Triplets 最终得分 100
用户昵称 123 运行时间 0.849 s
代码语言 C++ 内存使用 35.79 MiB
提交时间 2025-11-26 09:57:14
显示代码纯文本
#include <bits/stdc++.h>
using namespace std;
const int N=4e6+10; 
inline void read(int &x)
{
	int cnt=0;
	char c=getchar();
	while (!('0'<=c && c<='9')) c=getchar();
	while ('0'<=c && c<='9') cnt=cnt*10+(c-'0'),c=getchar();
	x=cnt;
}
int x,n,c,ans[10010],tot=0;
long long mp[N],a[N],b[N];
void solve1()
{
	long long ret=0;
	for (int i=0;i<c*2;i++) a[i]=a[(i-1 ? i-1 : 0)]+mp[i];
	for (int i=0;i<c*2;i++) b[i]=b[(i-1 ? i-1 : 0)]+mp[i]*a[i+c/2-1];
	for (int i=1;i<=c;i++) ret+=mp[i]*(b[i+c/2-1]-b[i]-a[i+c/2]*(a[i+c/2-1]-a[i]));
	cout<<ret/3; 
}
void solve2()
{
	long long ret=0;
	for (int i=0;i<c*2;i++) a[i]=a[(i-1 ? i-1 : 0)]+mp[i];
	for (int i=0;i<c*2;i++) b[i]=b[(i-1 ? i-1 : 0)]+mp[i]*a[i+c/2];
	for (int i=1;i<=c;i++) ret+=mp[i]*(b[i+c/2]-b[i]-a[i+c/2]*(a[i+c/2]-a[i]));
	cout<<ret/3;
}
int main() {
	freopen("Triplets.in","r",stdin);
	freopen("Triplets.out","w",stdout); 
	read(n),read(c);
	for (int i=1;i<=n;i++) read(x),mp[x]++,mp[x+c]++;
	if (c%2==0) solve1();
	else solve2();
	return 0;
}