ABC 342 D - Square Pair

题意：

       给一个序列a，求有多少个a[i]*a[j]是平方数（i<j）。

思路：

       本题的规律是两个完全平方数相乘也是一个完全平方数，也就是a[i]*a[j]可以表示为两个平方数的积。

       首先让输入的a[i]不能被比它小的平方数整除

        然后统计每个处理后的a[i]出现的次数，并计算出从a[1]到a[i]有多少个处理后不为0的数（本身就不为平方数的数）。

        接着按照题目的要求计算。

        最后把答案除以2再加上判断是否为0的结果。

代码：

#include <bits/stdc++.h>
using namespace std;
long long n,ans,num,a[200005],cnt[200005],b[200005];
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		for(int j=2;j*j<=a[i];j++){ //除以1没有意义 
			int jj=j*j;
			while(a[i]%jj==0) a[i]/=jj;
		}
		cnt[a[i]]++;
		b[i]=b[i-1]+(a[i]==0);
	}
	for(int i=1;i<=n;i++){
		if(a[i]==0) num+=n-b[i]; //判断是否为0 
		else ans+=cnt[a[i]]-1;
	}
	cout<<ans/2+num; //因为没有i<j，所以要除以2 
	return 0;
}