D. AND, OR and square sum(位运算,贪心)

题目传送门

题意: 给你n个数,对于i!=j,a[i]=x,a[j]=y,我们可以进行一种操作:a[i]=x&y,a[j]=x|y,你可以无限的进行这个操作,问你最后Σa[i] (1<=i<=n) 的最大值是多少。

思路: 我们每次贪心,每次拿出一个当前能拿出的最大的数,那肯定是用其他数和一个数取或运算,然后其他数就被他取与运算了,对于一个位上,有四种情况,(1|1 1&1) ,(1|0 ,1&0) ,(0|1,0&1), (0|0,0&0),我们发现在这个运算时,1的个数是不会变的所以我们要用每个位上有限的1的数量,去组合成n个数使得平方和最大。

贪心策略:前面我们知道了运算之后1的个数不会变,然后我们每次把还有剩余1的位都拿一个给一个数,这样从局部最优到整体最优。

代码:

#include<bits/stdc++.h>
#define endl '\n'
#define null NULL
#define ls p<<1
#define rs p<<1|1
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define ll long long
#define int long long
#define pii pair<int,int>
#define ull unsigned long long
#define all(x) x.begin(),x.end()
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define ct cerr<<"Time elapsed:"<<1.0*clock()/CLOCKS_PER_SEC<<"s.\n";
char *fs,*ft,buf[1<<20];
#define gc() (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<20,stdin),fs==ft))?0:*fs++;
inline int read(){int x=0,f=1;char ch=gc();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=gc();}
return x*f;}
using namespace std;
const int N=2e5+5;
const int inf=0x3f3f3f3f;
const int mod=1e9+7;
const double eps=1e-7;
const double PI=acos(-1);
int cnt[50],a[N];
signed main()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        int x;
        cin>>x;
        int p=0;
        while(x)
        {
            cnt[p]+=x%2;
            x>>=1;
            p++;
        }
    }
    int res=0;
    for(int i=1;i<=n;i++)
    {
        for(int j=0;j<=30;j++)
        {
            if(cnt[j])
            {
                a[i]+=(1<<j);
                cnt[j]--;
            }
        }
        res+=a[i]*a[i];
    }
    cout<<res<<endl;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章