【NOIP2010】【Vijos1775】乌龟棋

problem

solution

codes

#include<iostream>
using namespace std;
int n, m, a[355], b[5], dp[40][40][40][40];
int main(){
    cin>>n>>m;
    for(int i = 0; i < n; i++)cin>>a[i];
    for(int i = 0; i < m; i++){ int x; cin>>x; b[x]++;}
    dp[0][0][0][0] = a[0];
    for(int _1 = 0; _1 <= b[1]; _1++){
        for(int _2 = 0; _2 <= b[2]; _2++){
            for(int _3 = 0; _3 <= b[3]; _3++){
                for(int _4 = 0; _4 <= b[4]; _4++){
                                        //枚举最后一次用的哪一张。
                    int t = _1+_2*2+_3*3+_4*4;
                    if(_1)dp[_1][_2][_3][_4]=max(dp[_1][_2][_3][_4], dp[_1-1][_2][_3][_4]+a[t]);
                    if(_2)dp[_1][_2][_3][_4]=max(dp[_1][_2][_3][_4], dp[_1][_2-1][_3][_4]+a[t]);
                    if(_3)dp[_1][_2][_3][_4]=max(dp[_1][_2][_3][_4], dp[_1][_2][_3-1][_4]+a[t]);
                    if(_4)dp[_1][_2][_3][_4]=max(dp[_1][_2][_3][_4], dp[_1][_2][_3][_4-1]+a[t]);
                }
            }
        }
    }
    cout<<dp[b[1]][b[2]][b[3]][b[4]];
    return 0;
}