codeforces 999F. Cards and Joy

DP

題目傳送門

題目大意： 有$n$個人$n*k$張卡片，每個人都要分到$k$張卡片。卡片上有數字，每個人也有一個數字，當一個人分到$i$張和他的數字一樣的卡片時會有$h[i]$的貢獻，求最大貢獻。

設$g[i][j]$表示$i$張卡片分給$j$個人的最大貢獻，那麼有$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;
}