1. 程式人生 > >Codeforces 1077E (二分亂搞或者dp)

Codeforces 1077E (二分亂搞或者dp)

題意:給你一個數組,可以從中選區若干種元素,但每種元素選區的個數前一種必須是後一種的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);
}