[HDU 5074] Hatsune Miku (动态规划)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5074

题目大意是给你m个note，n个数，得分是v[a[i]][a[i+1]]的总和，如果说a[i]是负数的话代表可以放人一个note，否则就只能放他给的note号。

问：最大的得分是多少？

 

我先写了记忆化搜索函数

dp(i,j)代表到第i个位置，放标号为j的note

那么

如果说a[i+1]<0，那么dp(i,j) = max( dp(i,j) , dp(i+1,k)+v[j][k] );

否则dp(i,j) = dp(i+1,a[i+1]) + v[j][a[i+1]];

然后改成递推呗~

 

#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
using namespace std;

int T,n,m;
int v[55][55];
int dp[111][55];
int a[111];

int main(){
    scanf("%d",&T);
    while(T--){
        memset(v,0,sizeof(v));
        memset(dp,0,sizeof(dp));
        memset(a,0,sizeof(a));
        scanf("%d%d",&n,&m);
        for(int i=1;i<=m;i++){
            for(int j=1;j<=m;j++){
                scanf("%d",&v[i][j]);
            }
        }
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
        }
        for(int i=n-1;i>=0;i--){
            for(int j=0;j<=m;j++){
                if( a[i+1]<0 ){
                    for(int k=1;k<=m;k++){
                        dp[i][j] = max(dp[i][j],dp[i+1][k]+v[j][k]);
                    }
                } else {
                    dp[i][j] = v[j][a[i+1]]+dp[i+1][a[i+1]];
                }
            }
        }
        printf("%d\n",dp[0][0]);
    }
}