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;
}

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