There are nn players sitting at the card table. Each player has a favorite number. The favorite number of the jj-th player is fjfj.
There are k⋅nk⋅n cards on the table. Each card contains a single integer: the ii-th card contains number cici. Also, you are given a sequence h1,h2,…,hkh1,h2,…,hk. Its meaning will be explained below.
The players have to distribute all the cards in such a way that each of them will hold exactly kk cards. After all the cards are distributed, each player counts the number of cards he has that contains his favorite number. The joy level of a player equals htht if the player holds tt cards containing his favorite number. If a player gets no cards with his favorite number (i.e., t=0t=0), his joy level is 00.
Print the maximum possible total joy levels of the players after the cards are distributed. Note that the sequence h1,…,hkh1,…,hk is the same for all the players.
The first line of input contains two integers nn and kk (1≤n≤500,1≤k≤101≤n≤500,1≤k≤10) — the number of players and the number of cards each player will get.
The second line contains k⋅nk⋅n integers c1,c2,…,ck⋅nc1,c2,…,ck⋅n (1≤ci≤1051≤ci≤105) — the numbers written on the cards.
The third line contains nn integers f1,f2,…,fnf1,f2,…,fn (1≤fj≤1051≤fj≤105) — the favorite numbers of the players.
The fourth line contains kk integers h1,h2,…,hkh1,h2,…,hk (1≤ht≤1051≤ht≤105), where htht is the joy level of a player if he gets exactly tt cards with his favorite number written on them. It is guaranteed that the condition ht−1<htht−1<ht holds for each t∈[2..k]t∈[2..k].
Print one integer — the maximum possible total joy levels of the players among all possible card distributions.
4 3 1 3 2 8 5 5 8 2 2 8 5 2 1 2 2 5 2 6 7
21
3 3 9 9 9 9 9 9 9 9 9 1 2 3 1 2 3
0
In the first example, one possible optimal card distribution is the following:
- Player 11 gets cards with numbers [1,3,8][1,3,8];
- Player 22 gets cards with numbers [2,2,8][2,2,8];
- Player 33 gets cards with numbers [2,2,8][2,2,8];
- Player 44 gets cards with numbers [5,5,5][5,5,5].
Thus, the answer is 2+6+6+7=212+6+6+7=21.
In the second example, no player can get a card with his favorite number. Thus, the answer is 00.
我又被动态规划干倒了!
f[i][j]代表一共i张牌发给j个人能取得的最大价值。
#include<iostream>
#include<cstdio>
#include<string.h>
#include<algorithm>
#include<cmath>
using namespace std;
int n,k;
long long c[5010],a[505],h[15],f[5010][505],people[100010],card[100010];
int main()
{
scanf("%d%d",&n,&k);
memset(people,0,sizeof(people));
memset(card,0,sizeof(card));
for (int i=1; i<=n*k; i++)
{
scanf("%lld",&c[i]);
card[c[i]]++;
}
for (int i=1; i<=n; i++)
{
scanf("%lld",&a[i]);
people[a[i]]++;
}
for (int i=1; i<=k; i++)
{
scanf("%lld",&h[i]);
}
memset(f,0,sizeof(f));
for (int i=1; i<=n*k; i++)
{
f[i][1]=h[min(i,k)];
for (int j=2; j<=n; j++)
{
for (int u=1; u<=min(i,k); u++)
{
f[i][j]=max(f[i][j],f[i-u][j-1]+h[u]);
}
}
}
long long ans=0;
for (int i=0; i<=1e5; i++)
if (people[i])
ans+=f[card[i]][people[i]];
cout<<ans<<endl;
return 0;
}