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

时间:2023-03-09 04:01:31
[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[][];
int dp[][];
int a[]; int main(){
scanf("%d",&T);
while(T--){
memset(v,,sizeof(v));
memset(dp,,sizeof(dp));
memset(a,,sizeof(a));
scanf("%d%d",&n,&m);
for(int i=;i<=m;i++){
for(int j=;j<=m;j++){
scanf("%d",&v[i][j]);
}
}
for(int i=;i<=n;i++){
scanf("%d",&a[i]);
}
for(int i=n-;i>=;i--){
for(int j=;j<=m;j++){
if( a[i+]< ){
for(int k=;k<=m;k++){
dp[i][j] = max(dp[i][j],dp[i+][k]+v[j][k]);
}
} else {
dp[i][j] = v[j][a[i+]]+dp[i+][a[i+]];
}
}
}
printf("%d\n",dp[][]);
}
}