转载请注明出处:http://tokitsukaze.live/

题目链接:http://codeforces.com/problemset/problem/999/F


题意:有n*k张牌,每张牌上写了数字,分给n个人,每人k张。每个人有一个喜欢的数字,分到i张喜欢的牌,得到的价值就是h[i],问价值最大。


题解:
cnt[x]表示写着数字为x的牌的个数。
f[x]表示喜欢数字为x的牌的人数。

把cnt[x]看作背包容量,f[x]看作物品个数。对于每个物品,重量为i时,获得的价值为h[i],(0<=i<=k,h[0]=0)。

其实就是一个01背包。枚举x,做01背包就行了。

代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
#include <bits/stdc++.h>
#pragma comment(linker, "/STACK:1024000000,1024000000")
#define mem(a,b) memset((a),(b),sizeof(a))
#define MP make_pair
#define pb push_back
#define fi first
#define se second
#define sz(x) (int)x.size()
#define all(x) x.begin(),x.end()
using namespace std;
#define _GLIBCXX_PERMIT_BACKWARD_HASH
#include <ext/hash_map>
using namespace __gnu_cxx;
struct str_hash{size_t operator()(const string& str)const{return __stl_hash_string(str.c_str());}};
typedef long long ll;
typedef unsigned long long ull;
#define PII pair<int,int>
#define PLL pair<ll,ll>
#define PDD pair<double,double>
#define VI vector<int>
#define VL vector<ll>
const int INF=0x3f3f3f3f;
const ll LLINF=0x3f3f3f3f3f3f3f3f;
const double PI=acos(-1.0);
const double eps=1e-6;
const int MAX=1e5+10;
const ll mod=998244353;
int cnt[MAX],f[MAX],h[MAX],k;
ll dp[5050];
ll gao(int x)
{
int i,j,l;
if(!f[x]) return 0;
for(i=0;i<=cnt[x];i++) dp[i]=0;
while(f[x]--)
{
for(i=cnt[x];i;i--)
{
for(j=0;i-j>=0&&j<=k;j++)
{
dp[i]=max(dp[i],dp[i-j]+h[j]);
}
}
}
// cout<<x<<" "<<dp[cnt[x]]<<endl;
return dp[cnt[x]];
}
int main()
{
int n,i,x;
ll ans;
while(~scanf("%d%d",&n,&k))
{
mem(cnt,0);
mem(f,0);
for(i=1;i<=n*k;i++)
{
scanf("%d",&x);
cnt[x]++;
}
for(i=1;i<=n;i++)
{
scanf("%d",&x);
f[x]++;
}
h[0]=0;
for(i=1;i<=k;i++) scanf("%d",&h[i]);
ans=0;
for(i=1;i<=MAX-10;i++) ans+=gao(i);
printf("%lld\n",ans);
}
return 0;
}