AcWing 312. 四维计算公式
原题链接
简单
作者:
季之秋
,
2021-02-28 20:44:41
,
所有人可见
,
阅读 519
import java.util.*;
public class Main{
public static void main(String[] args){
Scanner sc=new Scanner(System.in);
int n=sc.nextInt();
int m=sc.nextInt();
int w[]=new int[n+1];
for(int i=0;i<n;i++) w[i]=sc.nextInt();
int cnt[]=new int[5];
while(m--!=0){
int a=sc.nextInt();
cnt[a]++;
}
int N=41;
int f[][][][]=new int[N][N][N][N];
for(int a=0;a<=cnt[1];a++){
for(int b=0;b<=cnt[2];b++){
for(int c=0;c<=cnt[3];c++){
for(int d=0;d<=cnt[4];d++){
f[a][b][c][d]=w[a+b*2+c*3+d*4];
if(a>=1) f[a][b][c][d]=Math.max(f[a][b][c][d],f[a-1][b][c][d]+w[a+b*2+c*3+d*4]);
if(b>=1) f[a][b][c][d]=Math.max(f[a][b][c][d],f[a][b-1][c][d]+w[a+b*2+c*3+d*4]);
if(c>=1) f[a][b][c][d]=Math.max(f[a][b][c][d],f[a][b][c-1][d]+w[a+b*2+c*3+d*4]);
if(d>=1) f[a][b][c][d]=Math.max(f[a][b][c][d],f[a][b][c][d-1]+w[a+b*2+c*3+d*4]);
}
}
}
}
System.out.println(f[cnt[1]][cnt[2]][cnt[3]][cnt[4]]);
}
}