AT_abc342_d 题解

UD 2024/2/24 22:36 感谢 Lixiang_is_potato 指出一处笔误。

本文同步发表于洛谷。

赛时挂了，但是赛后 3min AC，我是飞舞。

题意

给你一个长度为 \(N\) 的非负整数序列 \(A=(A_1,\ldots,A_N)\)。求满足以下两个条件的整数对 \((i,j)\) 的个数：

\(A_i A_j\) 是一个平方数且 \(1\leq i<j\leq n\)。

懒得提交翻译

思路

显然，\(A_i\) 的平方因子对他自己一点用都没有。

所以我们先给它的平方因子除掉，剩下的 \(A_i\) 中相等的两两配对。

但要注意 \(0\)，它可以和其他任何数相乘（\(0\) 也是平方数）！

代码

由于我的宏定义又臭又长我就删掉了。

LL n,a[200010],ans,sum,f[200010];
int main()
{
	cin>>n;
	rep(i,1,n,1)
	{
		cin>>a[i];
		for(LL j=2;j*j<=a[i];j++)//错
		{
			while(a[i]%(j*j)==0)
			{
				a[i]/=(j*j);
			}
		}
		f[a[i]]++;
		if(!a[i])ans++;
	}
	rep(i,0,200000,1)
	{
		sum+=f[i]*(f[i]-1)/2;
	}
	sum+=ans*(n-ans);
	cout<<sum<<endl;
	return 0;
}

小结

其实很感慨，赛时我代码里面打注释“错” 的那行我 \(j\) 的初值设成 \(1\) 了，调了好久… 导致直接掉分。

所以，衷心祝愿所有 OIer 能正常发挥，不犯太多错误，在比赛中取得好成绩！