【线性DP】乌龟棋

#include<bits/stdc++.h>

using namespace std;

int a[400],b[5];

int dp[50][50][50][50];

int main(){

    int n,m,t; 

    cin >> n >> m;

    for(int i=1;i<=n;i++){

        cin >> a[i];

    }

    for(int i=1;i<=m;i++){

        cin >> t;

        b[t]++;

    }

    for(int i=0;i<=b[1];i++){

        for(int j=0;j<=b[2];j++){

            for(int k=0;k<=b[3];k++){

                for(int l=0;l<=b[4];l++){

                    int dis=1+i*1+j*2+k*3+l*4;

                    int &v=dp[i][j][k][l];

                    v=a[dis];

                    if(i) v=max(v,dp[i-1][j][k][l]+a[dis]);

                    if(j) v=max(v,dp[i][j-1][k][l]+a[dis]);

                    if(k) v=max(v,dp[i][j][k-1][l]+a[dis]);

                    if(l) v=max(v,dp[i][j][k][l-1]+a[dis]);

                }

            }

        }

    }

    cout << dp[b[1]][b[2]][b[3]][b[4]] << '\n';

    return 0;

}