Codeforces Round #451 (Div. 2) E. Squares and not squares

题目大意

每次给一个数加一或减一，问最小操作次数使一半是完全平方数一半不是。

题解

显然只需要将多的一部分变成少的一部分，贪心选需要操作次数最少的。

#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#include<cmath>using namespace std;int read(){    char ch=getchar();int f=0;    while(ch<'0'||ch>'9') ch=getchar();    while(ch>='0'&&ch<='9') {f=f*10+(ch^48);ch=getchar();}    return f;}struct data{    int num;    int step;    int type;}a[500005];bool cmp(data x,data y){    if(x.type==y.type)    {        return x.step<y.step;    }    else return x.type<y.type;}int n;int main(){    n=read();    for(int i=1;i<=n;i++)    {        a[i].num=read();        int temp=sqrt(a[i].num);        if(temp*temp==a[i].num)        {            a[i].type=0;            if(a[i].num==0) a[i].step=2;            else a[i].step=1;        }        else        {            a[i].type=1;            a[i].step=min(a[i].num-temp*temp,(temp+1)*(temp+1)-a[i].num);        }    }    int pla=0;    sort(a+1,a+n+1,cmp);    for(int i=1;i<=n;i++)    {        if(a[i].type==1)        {            pla=i-1;            break;        }    }    if(a[n].type==0) pla=n;    int pla2=n-pla;    //cout<<pla<<" "<<pla2<<endl;    if(pla>pla2)    {        long long ans=0;        for(int i=1;i<=n;i++)        {            ans+=a[i].step;            pla--;pla2++;            if(pla==pla2) break;        }        cout<<ans;    }    else if(pla<pla2)    {        long long ans=0;        for(int i=pla+1;i<=n;i++)        {            ans+=a[i].step;            pla++;pla2--;            if(pla==pla2) break;        }        cout<<ans;    }    else cout<<"0";}