题目
题意:
给出n(n<=300)个pair:
<key,value>
<script type="math/tex" id="MathJax-Element-1">
</script>
如果相邻两个pair的key不互素,那么可以将这两个pair拿取,得分加上两个value。
问最大得分。
解:
很显然就是区间dp,一开始我认为就是枚举一个区间中最后拿去的两个相邻pair就完了,wa一发后,发现不对:
最后拿去的两个pair不一定相邻,所以如果[le,ri]中le和k配对,那么[le,k]必然能够全部拿去,之后计算DP(k+1,ri)
,所以需要预处理哪些段能够全部拿去,具体方法是枚举le和谁配对。
比赛代码:
#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<vector>
using namespace std;
#define all(x) (x).begin(), (x).end()
#define for0(a, n) for (int (a) = 0; (a) < (n); (a)++)
#define for1(a, n) for (int (a) = 1; (a) <= (n); (a)++)
#define mes(a,x,s) memset(a,x,(s)*sizeof a[0])
#define mem(a,x) memset(a,x,sizeof a)
#define ysk(x) (1<<(x))
typedef long long ll;
typedef pair<int, int> pii;
const int INF =0x3f3f3f3f;
const int maxn= 300 ;
bool can[maxn+5][maxn+5];
int n;
ll dp[maxn+4][maxn+4];
int key[maxn+5],val[maxn+5];
bool take[maxn+4][maxn+4];
ll sum[maxn+4];
bool TAKE(int le,int ri)
{
if(le>ri) return true;
else if(le==ri) return false;
return take[le][ri];
}
void solve()
{
for1(i,n)
{
for(int j=i+1;j<=n;j++)
{
if(__gcd(key[i],key[j])==1 ) can[i][j]=0;
else can[i][j]=1;
}
}
for1(i,n) take[i][i]=0;
for(int add=1;add<n;add++)
{
for(int le=1;le+add<=n;le++)
{
int ri=le+add;
take[le][ri]=0;
for(int k=le+1;k<=ri;k++)
{
if(can[le][k]&& TAKE(le+1,k-1)&&TAKE(k+1,ri) ) take[le][ri]=1;
}
}
}
}
ll DP(int le,int ri)
{
if(le>=ri) return 0;
if(~dp[le][ri]) return dp[le][ri];
ll &ans=dp[le][ri];
ans=0;
for(int k=le+1;k<=ri;k++)
{
if(take[le][k])
{
ans=max(ans,DP(k+1,ri)+sum[k]-sum[le-1] );
}
}
ans=max(ans,DP(le+1,ri) );
return ans;
}
int main()
{
std::ios::sync_with_stdio(false);
int T;cin>>T;
while(T--)
{
cin>>n;
for1(i,n) cin>>key[i];
sum[0]=0;
for1(i,n)
{
cin>>val[i];
sum[i]=sum[i-1]+val[i];
}
solve();
mem(dp,-1);
printf("%lld\n",DP(1,n));
}
return 0;
}
/*
123
3
1 2 4
1 1 1
*/