codeforces 999F. Cards and Joy

DP

題目傳送門

題目大意:nn個人nkn*k張卡片,每個人都要分到kk張卡片。卡片上有數字,每個人也有一個數字,當一個人分到ii張和他的數字一樣的卡片時會有h[i]h[i]的貢獻,求最大貢獻。

g[i][j]g[i][j]表示ii張卡片分給jj個人的最大貢獻,那麼有g[i][j]=max{g[ip][j1]+h[p]}g[i][j]=max\{g[i-p][j-1]+h[p]\}。對於每個人的數字算一下就好了。

代碼:

#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100005
#define M 505
#define F inline
using namespace std;
int n,k,mx,ans,c[N],f[N],h[11],g[M*10][M];
F char readc(){
	static char buf[100000],*l=buf,*r=buf;
	if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
	return l==r?EOF:*l++;
}
F int _read(){
	int x=0; char ch=readc();
	while (!isdigit(ch)) ch=readc();
	while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=readc();
	return x;
}
int main(){
	n=_read(),k=_read();
	for (int i=1;i<=n*k;i++) c[_read()]++;
	for (int i=1,x;i<=n;i++) mx=max(mx,x=_read()),f[x]++;
	for (int i=1;i<=k;i++) h[i]=_read();
	for (int j=1;j<=n;j++)
	for (int i=1;i<=j*k;i++)
	for (int p=1;p<=k;p++)
		if (i>=p) g[i][j]=max(g[i][j],g[i-p][j-1]+h[p]);
	for (int i=1;i<=mx;i++)
		if (f[i]) ans+=g[min(f[i]*k,c[i])][f[i]];
	return printf("%d\n",ans),0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章