F - Cards and Joy
题 意: n个人,每个人k张牌,每张牌都写着一个数字ci,每个人都有一种喜欢的牌fi。当一个人手中有k张自己喜欢的牌的时候那么它的开心程度是h[k],问每个玩家最大的开心程度。
数据范围:
1<=n<=500
1<=k<=50
1<=ci<=1e5
1<=fi<=1e5
1<=ht<=1e5
输入样例:
4 3
1 3 2 8 5 5 8 2 2 8 5 2
1 2 2 5
2 6 7
输出样例:
21
思 路:经典dp题,dp[i][j]表示i张相同的牌发给j个喜欢这张牌的人的最大价值。
那么dp[i][j] = dp[i-u][j-1]+h[u],每个人只能喜欢一张牌,而且欢乐程度,也之和喜欢的这张牌有关num[i],代表数组为i的牌的个数,ans[i],代表喜欢这张牌的人数。
那么最后dp[num[i]][ans[i]] 就是答案。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 5e2+5;
const int MAX = 1e5+5;
int dp[5000+5][maxn];
int c[5000+5],f[maxn];
int h[maxn];
int num[MAX],ans[MAX];
int n,k;
//dp[i][j];i种相同的牌分给j个人的最大价值
//dp[i][j] = dp[i-u][j-u] + h[u];
int main(){
scanf("%d %d",&n,&k);
for(int i=0;i<k*n;i++) {
scanf("%d",&c[i]);
num[c[i]]++;
}
for(int i=0;i<n;i++){
scanf("%d",&f[i]);
ans[f[i]]++;
}
for(int i=1;i<=k;i++){
scanf("%d",&h[i]);
}
for(int i=1;i<=k*n;i++){
dp[i][1] = h[min(i,k)];
for(int j=2;j<=n;j++){
for(int u=1;u<=min(i,k);u++){
dp[i][j] = max(dp[i][j],dp[i-u][j-1]+h[u]);
}
}
}
ll ans1 = 0;
for(int i=1;i<=MAX-5;i++)if(num[i]){
ans1+= dp[num[i]][ans[i]];
}
printf("%lld\n",ans1);
return 0;
}