wikioi 1068 乌龟棋

四维dp表示分别用了多少张卡后可以获得的最大分数

#include <cstdio>
#include <cmath>
#include <cstring>
#include <ctime>
#include <iostream>
#include <algorithm>
#include <set>
#include <vector>
#include <sstream>
#include <queue>
#include <typeinfo>

typedef long long ll;
using namespace std;
int a[400];
int b[5];
int dp[50][50][50][50];
int main()
{
    memset(b,0,sizeof(b));
    int n,m;
    cin>>n>>m;
    for(int i=0;i<n;i++)
        cin>>a[i];
    for(int i=0;i<m;i++)
    {
        cin>>b[0];
        b[b[0]]++;
    }
    int ans;
    dp[0][0][0][0]=a[0];
    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 m=0;m<=b[4];m++)
                {
                    ans=0;
                    if(i)ans=max(ans,dp[i-1][j][k][m]);
                    if(j)ans=max(ans,dp[i][j-1][k][m]);
                    if(k)ans=max(ans,dp[i][j][k-1][m]);
                    if(m)ans=max(ans,dp[i][j][k][m-1]);
                    dp[i][j][k][m]=ans+a[i+2*j+3*k+4*m];
                }
            }
        }
    }
    cout<<dp[b[1]][b[2]][b[3]][b[4]]<<endl;
    return 0;
}