记f[a][b][c][d]为四张牌分别用了a,b,c,d张时的最大得分,然后可以递推
代码
#include<bits/stdc++.h>
using namespace std;
int n,m,x,a[355],f[125][125][125][125],col[5];
int main()
{
// freopen("a.in","r",stdin);
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
for(int i=1;i<=m;i++) scanf("%d",&x),col[x]++;
f[0][0][0][0]=a[1];
int now,maxx;
for(int aa=0;aa<=col[1];aa++)
for(int b=0;b<=col[2];b++)
for(int c=0;c<=col[3];c++)
for(int d=0;d<=col[4];d++)
{
if(!aa&&!b&&!c&&!d) continue;
maxx=0; now=1+aa+2*b+3*c+4*d;
if(aa) maxx=max(maxx,f[aa-1][b][c][d]+a[now]);
if(b) maxx=max(maxx,f[aa][b-1][c][d]+a[now]);
if(c) maxx=max(maxx,f[aa][b][c-1][d]+a[now]);
if(d) maxx=max(maxx,f[aa][b][c][d-1]+a[now]);
f[aa][b][c][d]=max(maxx,f[aa][b][c][d]);
}
printf("%d\n",f[col[1]][col[2]][col[3]][col[4]]);
return 0;
}