感觉像变态的背包
状态转移方程为:f(i,j,k,l)=max{
f(i-1,j,k,l)+a[(i-1)*1+j*2+k*3+l*4+2],
f(i,j-1,k,l)+a[i*1+(j-1)*2+k*3+l*4+3],
f(i,j,k-1,l)+a[i*1+j*2+(k-1)*3+l*4+4],
f(i,j,k,l-1)+a[i*1+j*2+k*3+(l-1)*4+5]
}
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=350+10;
const int maxm=40+10;
int n,m,a[maxn],f[maxm][maxm][maxm][maxm],c[5];
int step(int i,int j,int k,int l)
{
return i+j*2+k*3+l*4+1;
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
for(int i=1;i<=m;i++)
{
int b;
scanf("%d",&b);
c[b]++;
}
f[0][0][0][0]=a[1];
for(int i=0;i<=c[1];i++)
for(int j=0;j<=c[2];j++)
for(int k=0;k<=c[3];k++)
for(int l=0;l<=c[4];l++)
{
if(i>=1)
f[i][j][k][l]=max(f[i][j][k][l],f[i-1][j][k][l]+a[step(i-1,j,k,l)+1]);
if(j>=1)
f[i][j][k][l]=max(f[i][j][k][l],f[i][j-1][k][l]+a[step(i,j-1,k,l)+2]);
if(k>=1)
f[i][j][k][l]=max(f[i][j][k][l],f[i][j][k-1][l]+a[step(i,j,k-1,l)+3]);
if(l>=1)
f[i][j][k][l]=max(f[i][j][k][l],f[i][j][k][l-1]+a[step(i,j,k,l-1)+4]);
}
printf("%d\n",f[c[1]][c[2]][c[3]][c[4]]);
return 0;
}
#include<algorithm>
using namespace std;
const int maxn=350+10;
const int maxm=40+10;
int n,m,a[maxn],f[maxm][maxm][maxm][maxm],c[5];
int step(int i,int j,int k,int l)
{
return i+j*2+k*3+l*4+1;
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
for(int i=1;i<=m;i++)
{
int b;
scanf("%d",&b);
c[b]++;
}
f[0][0][0][0]=a[1];
for(int i=0;i<=c[1];i++)
for(int j=0;j<=c[2];j++)
for(int k=0;k<=c[3];k++)
for(int l=0;l<=c[4];l++)
{
if(i>=1)
f[i][j][k][l]=max(f[i][j][k][l],f[i-1][j][k][l]+a[step(i-1,j,k,l)+1]);
if(j>=1)
f[i][j][k][l]=max(f[i][j][k][l],f[i][j-1][k][l]+a[step(i,j-1,k,l)+2]);
if(k>=1)
f[i][j][k][l]=max(f[i][j][k][l],f[i][j][k-1][l]+a[step(i,j,k-1,l)+3]);
if(l>=1)
f[i][j][k][l]=max(f[i][j][k][l],f[i][j][k][l-1]+a[step(i,j,k,l-1)+4]);
}
printf("%d\n",f[c[1]][c[2]][c[3]][c[4]]);
return 0;
}