Wannafly挑戰賽27——A.灰魔法師
阿新 • • 發佈:2018-11-02
坑爹的簽到題,差點做不出來
給一個數組,求其中兩個數相加為完全平方數的對數
n是1e5
暴力會超時的,所以先預處理出2*1e5內的完全平方數,然後每輸入一個數,就標記下它需要和哪個數相加才能湊成完全平方數,然後後面的每讀入一個除了要記錄,還要看這個數是否前面已經被記錄過,就是加上它就能湊成完全平方數,直接加上num[a[i]],然後還要開long long
#include <bits/stdc++.h>
using namespace std;
int n;
const int N=1e6+50;
int a[N];
int Pow[N];
int num[N];
int main(void){
int k=0;
for(int i=2;i*i<N;i++){
Pow[k++]=i*i;
//printf("%d %d\n",i,Pow[k-1]);
}
long long cnt=0;
scanf("%d",&n);
for(int i=0;i<n;i++){
scanf("%d",&a[i]);
if(num[a[i]]){
cnt+=num[a[i]];
}
for(int j=0;j<k;j++){
num[Pow[j]-a[i]]++;
}
}
printf("%lld\n",cnt);
return 0;
}