AcWing 312/NOIP2010 乌龟棋(动态规划,线性dp)

传送门
题目大意:给了一个权值序列,一开始人在1处,有1,2,3,4四种卡牌若干张使用后移动对应点数.每张卡牌只能用1次,移动后会得到对应点的权值.求能得到的权值最大和.
标准dp问题,用一个dp[i][a][b][c][d]来表示移动到i的时候四种卡牌使用了a,b,c,d张得到的max.
MLE.
因为四种卡牌是刚刚好能移动到终点的,所以第一维i可以用a,b,c,d推导出来.i = a+2b+3c+4d+1. +1是因为一开始在点1.
用刷表法做的话转移方程为:
dp[a+1][b][c][d] = max(dp[a+1][b][c][d],dp[a][b][c][d]+w[i+1]);
dp[a][b+1][c][d] = max(dp[a][b+1][c][d],dp[a][b][c][d]+w[i+2]);
c+1,d+1同理.
代码:

#pragma GCC optimize(2)
#define LL long long
#define pq priority_queue
#define ULL unsigned long long
#define pb push_back
#define mem(a,x) memset(a,x,sizeof a)
#define pii pair<int,int>
#define fir(i,a,b) for(int i=a;i<=(int)b;++i)
#define afir(i,a,b) for(int i=(int)a;i>=b;--i)
#define ft first
#define vi vector<int>
#define sd second
#define ALL(a) a.begin(),a.end()
#define bug puts("-------")
#define mpr(a,b) make_pair(a,b)
#include <bits/stdc++.h>

using namespace std;
const int N = 40+10;

inline void read(int &a){
    int x = 0,f=1;char ch = getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=getchar();}
    a = x*f;
}
int card,num[4],n,m,dp[N][N][N][N],w[N*10];
int main(){
    cin >> n >> m;
    fir(i,1,n) cin >> w[i];
    fir(i,1,m) cin >> card,num[card-1]++;
    dp[0][0][0][0] = w[1];
    fir(a,0,num[0]){
        fir(b,0,num[1]){
            fir(c,0,num[2]){
                fir(d,0,num[3]){
                    int i = a+b*2+c*3+d*4+1;
                    dp[a+1][b][c][d] = max(dp[a+1][b][c][d],dp[a][b][c][d]+w[i+1]);
                    dp[a][b+1][c][d] = max(dp[a][b+1][c][d],dp[a][b][c][d]+w[i+2]);
                    dp[a][b][c+1][d] = max(dp[a][b][c+1][d],dp[a][b][c][d]+w[i+3]);
                    dp[a][b][c][d+1] = max(dp[a][b][c][d+1],dp[a][b][c][d]+w[i+4]);
                }
            }
        }
    }
    cout << dp[num[0]][num[1]][num[2]][num[3]];
    
    return 0;
}