1. 程式人生 > >Wannafly挑戰賽27——A.灰魔法師

Wannafly挑戰賽27——A.灰魔法師

坑爹的簽到題,差點做不出來
給一個數組,求其中兩個數相加為完全平方數的對數
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; }