给定一个序列 有些位数未知,给你所有两个数连续所得到的能量,问你怎么安排数字使得总能量最大
二维dp,dp[i][j]表示第i位放音符j
分类讨论即可
#include#include #include #include #include #include #include #include #include using namespace std;#define RD(x) scanf("%d",&x)#define RD2(x,y) scanf("%d%d",&x,&y)#define RD3(x,y,z) scanf("%d%d%d",&x,&y,&z)#define clr0(x) memset(x,0,sizeof(x))typedef long long LL;const int maxn = 105;int n,m,s[maxn][maxn],p[maxn];int dp[maxn][maxn];//i位,此位jint main() { int _;RD(_);while(_--){ RD2(n,m); for(int i = 1;i <= m;++i) for(int j = 1;j <= m;++j){ RD(s[i][j]); } for(int i = 1;i <= n;++i) RD(p[i]); clr0(dp); for(int i = 1;i < n;++i){ if(p[i] > 0){ if(p[i+1] > 0) dp[i+1][p[i+1]] = dp[i][p[i]] + s[p[i]][p[i+1]]; else for(int j = 1;j <= m;++j) dp[i+1][j] = max(dp[i+1][j],dp[i][p[i]] + s[p[i]][j]); } else{ if(p[i+1] > 0) for(int j = 1;j <= m;++j) dp[i+1][p[i+1]] = max(dp[i+1][p[i+1]],dp[i][j] + s[j][p[i+1]]); else for(int j = 1;j <= m;++j) for(int k = 1;k <= m;++k) dp[i+1][j] = max(dp[i+1][j],dp[i][k] + s[k][j]); } } int ans = 0; for(int i = 1;i <= m;++i) ans = max(ans,dp[n][i]); printf("%d\n",ans); } return 0;}