题目大意:给你n对pair 里面有两个值,分别是key 和 val 。你可以取相邻的两个pair 获得其中的val,前提是两个pair 的key 的 gcd 不为 1。当然你把相邻的两个取走了之后原本不相邻的两个就变得相邻了。比如:你将下标为 2,3 取走之后,下标1,4就变得相邻了,求你可以获得的最大val。
这是2016网络赛的签到题吧,可是做了一个下午都没有做出来。当时写的那个代码现在都不知道哪里有错,各种测试数据都试了。后来看了大神的代码还有解释才懂的。
解题思路:这是一个典型的区间dp,也是我的第一道dp题。
dp[i][j] 表示 i 到 j 的最大值。考虑区间【i,j】的情况。
1,当gcd(key[i],key[j]) > 1 时,如果j = i+1,则dp[i][j] = val[i] + val[j]。否则要将【i+1,j-1】内全部去掉才可以。dp[i][j] = max(dp[i][j],dp[i+1][j-1] + val[i] +val[j])。
2,想知道【i+1,j-1】是不是能够去掉只需要用一个前缀和来判断就可以了。当dp[i+1][j-1] == sum[j-1] = sum[i] 就表示可以全部去掉了。。
3,如果区间【i,j】不能合并就枚举区间【i,j】中间的数k。找出区间内满足情况的最大值。
代码:
#include <bits/stdc++.h>
#define N 330
#define INF (1ll << 60)-1
#define LL long long
using namespace std;
LL dp[N][N];
int key[N],val[N];
LL sum[N]; // 前缀和数组
int T,n;
int GCD(int a,int b){
return (b == 0)?a:GCD(b,a%b);
}
LL DP(){
memset(dp,0,sizeof(dp));
for (int l = 1;l <= n;++l){
for (int i = 1;i+l <= n;++i){
int j = i + l;
for (int k = i;k < j;++k) dp[i][j] =max(dp[i][j], dp[i][k] + dp[k+1][j]); // 枚举k
if (GCD(key[i],key[j]) > 1){
//判断i和j是不是相邻,如果不相邻则判断区间能否消除。
if (j == i+1) dp[i][j] = val[i] + val[j];
else if (dp[i+1][j-1] == sum[j-1]-sum[i]) dp[i][j] = dp[i+1][j-1] + val[i] + val[j];
}
}
}
return dp[1][n];
}
int main(){
cin >> T;
while(T--){
memset(sum,0,sizeof(sum));
cin >> n;
for (int i = 1;i <= n;++i) scanf("%d",&key[i]);
for (int i = 1;i <= n;++i) {scanf("%d",&val[i]);sum[i] = val[i] + sum[i-1];}
printf("%lld\n",DP());
}
return 0;
}