SCU 4510 TaoSama与煎饼（dp）

题目传送门：http://acm.scu.edu.cn/soj/problem.action?id=4510
有120个道具，350个加工的地方，每个地方可以提高美味度，然后问你怎么使用道具，可以使停留的地方的美味度加和最大
考虑到就只有4种道具，并且每个道具就40个最多，然后我们想到转移肯定是现在走了几个地方，每种道具还剩多少个，但是这样会MLE，所以需要压缩状态，考虑到其实只有3个道具的个数是需要的，还有一个可以通过另外3个算出来，因为总数确定，并且可以不用记录走到了哪里，可以直接记录总的道具还剩多少个，因为走到哪里可以通过使用了多少道具算出来，所以最后的状态为 dp[120][40][40][40] ，这样压缩就可以过了

代码：

#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <string>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <sstream>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#pragma comment(linker,"/STACK:102400000,102400000")

using namespace std;
#define MAX 200005
#define MAXN 1000005
#define maxnode 10
#define sigma_size 2
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define lrt rt<<1
#define rrt rt<<1|1
#define middle int m=(r+l)>>1
#define LL long long
#define ull unsigned long long
#define mem(x,v) memset(x,v,sizeof(x))
#define lowbit(x) (x&-x)
#define pii pair<int,int>
#define bits(a) __builtin_popcount(a)
#define mk make_pair
#define limit 10000

//const int prime = 999983;
const int    INF   = 0x3f3f3f3f;
const LL     INFF  = 0x3f3f;
const double pi    = acos(-1.0);
const double inf   = 1e18;
const double eps   = 1e-9;
const LL     mod   = 1e9+7;
const ull    mxx   = 1333331;

/*****************************************************/
inline void RI(int &x) {
      char c;
      while((c=getchar())<'0' || c>'9');
      x=c-'0';
      while((c=getchar())>='0' && c<='9') x=(x<<3)+(x<<1)+c-'0';
}
/*****************************************************/

int d[121][41][41][41];
int a[351];
int num[5];

int dp(int all,int n1,int n2,int n3){
    if(d[all][n1][n2][n3]!=-1) return d[all][n1][n2][n3];
    int ans=0;
    if(all==0) return a[1];
    if(n1){
        int temp=dp(all-1,n1-1,n2,n3);
        ans=max(ans,temp);
    }
    if(n2){
        int temp=dp(all-1,n1,n2-1,n3);
        ans=max(ans,temp);
    }
    if(n3){
        int temp=dp(all-1,n1,n2,n3-1);
        ans=max(ans,temp);
    }
    if(all-n1-n2-n3>0){
        int temp=dp(all-1,n1,n2,n3);
        ans=max(ans,temp);
    }
    return d[all][n1][n2][n3]=ans+a[n1+2*n2+3*n3+(all-n1-n2-n3)*4+1];
}
int main(){
    //freopen("in.txt","r",stdin);
    int t;
    cin>>t;
    while(t--){
        int n,m;
        cin>>n>>m;
        mem(num,0);
        for(int i=1;i<=n;i++) RI(a[i]);
        for(int i=0;i<m;i++){
            int b;
            RI(b);
            num[b]++;
        }
        mem(d,-1);
        cout<<dp(m,num[1],num[2],num[3])<<endl;
    }
    return 0;
}