CF1081E Missing Numbers

初中数学

 

首先记p1^2,p2^2...pn^2,为构造完的a数组的前缀和（如果存在的话），假定已经处理好a数组的前 i - 2 项，现在正在处理第 i - 1 项和第 i 项（其中i为偶数）

 

然后得出 ai = pi ^ 2 - (pi-1) ^ 2，再用初中学过的因式分解处理一下，就得到ai = (pi + pi-1) * (pi - pi-1)

 

注意到 ai <= 1e5 ,所以可以直接枚举ai的因数得到 (pi + pi-1) 和 (pi - pi-1)，再解一次二元一次方程组，就能得到一些组 pi 和 pi-1

 

那么这些组pi和pi-1，哪些是有用的呢

 

首先pi代表了正数数组的前缀和,那它一定是递增的，所以对于小于pi-2的pi-1就是不合法的

 

那么按照贪心的想，为了让pi后面的选择更多，就得选取pi尽可能的小

 

这样我们就把p数组求出来了，然后求差就能得到a数组

 

#include<bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define db double
#define rep(x,a,b) for(int x=(a);x<=(b);x++)
#define per(x,a,b) for(int x=(a);x>=(b);x--)
#define reP(x,a,b) for(int x=(a);x<(b);x++)
#define Per(x,a,b) for(int x=(a);x>(b);x--)
#define scf(a) scanf("%d",&a)
#define scfll(a) scanf("%lld",&a)
#define scfdb(a) scanf("%lf",&a)
#define ptf(a) printf("%d",a)
#define ptfll(a) printf("%lld",a)
#define ptfdb(x,a) printf("%.xlf",a)
#define ptfsp(a) printf("%d ",a)
#define ptfllsp(a) printf("%lld ",a)
#define ptfdbsp(x,a) printf("%.xlf ",a)
#define pli(a,b) make_pair(a,b)
#define pb push_back
#define el puts("")
#define FS first
#define SE second
#define PI acos(-1)
/*ios::sync_with_stdio(false);
freopen("in.txt","r",stdin);
freopen("1.out","w",stdout);*/
using namespace std;
const ll mod=998244353;
const int maxn=2e5+5;
ll p[maxn],a[maxn];
int main(){
    int n;scf(n);
    rep(i,1,n) p[i]=1e18;
    rep(i,1,n/2){
        int x;scf(x);
        bool flag=0;
        for(int j=1;j*j<=x;j++){
            if(x%j) continue;
            int nj=x/j;
            if(j==nj||(j+nj)%2||(nj-j)%2) continue;
            ll P=(j+nj)/2,Q=(nj-j)/2;
           // cout<<i<<" "<<P<<" "<<Q<<endl;
            if(Q*Q<=p[i*2-2]) continue;
            flag=1;
            if(p[i*2]>P*P){
                p[i*2]=P*P;
                p[i*2-1]=Q*Q;
            }
        }
        if(!flag){
            puts("No");
            return 0;
        }
    }
    puts("Yes");
    rep(i,1,n) ptfllsp(p[i]-p[i-1]);
}