显示代码纯文本
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int n,m,sn[351],sm[5],f[41][41][41][41],te;
void hs(int a,int b,int c,int d)
{
if(a!=0) f[a][b][c][d]=max(f[a][b][c][d],f[a-1][b][c][d]+sn[a+b*2+c*3+d*4+1]);
if(b!=0) f[a][b][c][d]=max(f[a][b][c][d],f[a][b-1][c][d]+sn[a+b*2+c*3+d*4+1]);
if(c!=0) f[a][b][c][d]=max(f[a][b][c][d],f[a][b][c-1][d]+sn[a+b*2+c*3+d*4+1]);
if(d!=0) f[a][b][c][d]=max(f[a][b][c][d],f[a][b][c][d-1]+sn[a+b*2+c*3+d*4+1]);
return;
}
int main()
{
freopen("tortoise.in","r",stdin);
freopen("tortoise.out","w",stdout);
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
scanf("%d",&sn[i]);
for(int i=1;i<=m;i++)
{
scanf("%d",&te);
sm[te]++;
}
f[0][0][0][0]=sn[1];
for(int t1=0;t1<=sm[1];t1++)
for(int t2=0;t2<=sm[2];t2++)
for(int t3=0;t3<=sm[3];t3++)
for(int t4=0;t4<=sm[4];t4++)
hs(t1,t2,t3,t4);
cout<<f[sm[1]][sm[2]][sm[3]][sm[4]]<<endl;;
return 0;
}