Codeforces 1077E (二分亂搞或者dp)
阿新 • • 發佈:2018-11-17
題意:給你一個數組,可以從中選區若干種元素,但每種元素選區的個數前一種必須是後一種的2倍,選區的任意2種元素不能相同,問可以選取最多的元素個數是多少?
思路1(亂搞):記錄一下每種元素的個數,然後暴力列舉最少的元素個數,計算符合題意的最優情況。
程式碼:
#include<cstdio> #include<algorithm> #include<cmath> #include<cstring> #include<map> #include<set> #include<bitset> #include<vector> #include<iostream> #define INF 0x3f3f3f3f #define LL long long #define fi first #define se second #define mk make_pair #define pb push_back using namespace std; const int maxn=1000010; map<int,int> mp; map<int,int>::iterator it; int a[maxn],b[maxn],cnt; int v[maxn]; int main(){ int n,ans=0; // freopen("in.txt","r",stdin); scanf("%d",&n); for(int i=1;i<=n;i++){ scanf("%d",&a[i]); mp[a[i]]++; } for(it=mp.begin();it!=mp.end();it++){ b[++cnt]=it->second; } sort(b+1,b+1+cnt); for(int i=1;i<=b[cnt];i++){ int now=0,sum=i; while(now<=n){ int pos=lower_bound(b+1,b+1+cnt,sum)-b; while(pos<=cnt&&v[pos]==i)pos++; v[pos]=i; if(pos>cnt)break; now+=sum; sum=sum+sum; } ans=max(ans,now); } printf("%d\n",ans); }
思路2:DP 設dp[i]為最少元素個數為i時的最優解,將元素的個數從小到大排序後,從後往前更新答案。
程式碼:
#include<bits/stdc++.h> using namespace std; const int maxn=200010; int dp[maxn*2]; map<int,int> mp; int a[maxn]; int main(){ int n,ans=0,cnt=0; scanf("%d",&n); for(int i=1;i<=n;i++){ scanf("%d",&a[i]); mp[a[i]]++; } for(auto i:mp){ a[++cnt]=i.second; } sort(a+1,a+1+cnt); for(int i=cnt;i>=1;i--) for(int j=1;j<=a[i];j++){ dp[j]=max(dp[j],j+dp[j*2]); ans=max(ans,dp[j]); } printf("%d\n",ans); }