题目描述
一天,神犇和 LCR 在玩扑克牌。他们玩的是一种叫做“接竹竿”的游戏。
游戏规则是:一共有 n 张牌,每张牌上有一个花色 c 和一个点数v,花色不超过 k 种。将这些牌依次放入一列牌的末端。若放入之前这列牌中已有与这张牌花色相同的牌,你可以选择将这张牌和任意一张花色相同的牌之间的所有牌全部取出队列(包括这两张牌本身),并得到与取出的所有牌点数和相同的分数。现在已知 LCR 把这 n 张牌放入队列的顺序,求她最多能得多少分。
输入顺序即为 LCR 放入队列的顺序。即,ci表示第 i 张放入的牌的花色,vi表示第 i 张放入的牌的点数。
加粗样式
请注意,如果你知道类似的纸牌游戏,请尤其仔细地阅读规则,以免因为理解题意错误而出现不必要的问题。
输入格式
第一行两个整数 n,k。
第二行,n 个整数 c1,c2,…,cn表示花色,满足 1≤ci≤k
第三行,n 个整数 v1,v2,…,vn表示点数
输出格式
输出一行一个整数,表示最多能得到的分数。
样例输入 1
7 3
1 2 1 2 3 2 3
1 2 1 2 3 2 3
样例输出 1
13
样例输入 2
18 5
5 2 3 5 1 3 5 2 1 4 2 4 5 4 1 1 1 5
8 2 7 6 10 8 10 9 10 2 4 7 7 7 7 9 7 3
样例输出 2
123
思路
一个弱弱的DP
DP方程即枚举前面一个与这张牌花色一样的牌求出最大值
易得DP方程:dp[i]=max(dp[i],dp[j-1]+sum[i]-sum[j-1]);其中c[i]=c[j];
显然这样是一个O(N^2)时间
然而仔细观察DP数组 发现dp[j-1]和sum[j-1]的下标都是j-1
所以可以将这两个变量放在q数组里面每次比较最大值
于是
dp[i]=max(q[c[i]]+sum[i],dp[i-1]);
q[c[i]]=max(q[c[i]],dp[i-1]-sum[i-1]);//注意:是将dp[i-1]-sum[i-1]放在q数组里而不是dp[i]-sum[i]
好的
然后
就是
一个剧蒟蒻的线型DP
上代码:
哈哈哈
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define rep(i,a,b) for(long long i=a;i<=b;i++)
using namespace std;
typedef long long ll;
ll n,m,dp[1000050],q[1000050],c[1000050],v[1000050],sum[1000050],ans=0; //q[i]表示颜色i可以取之前最大的值,即dp[j]-sum[j]
int main()
{
memset(q,128,sizeof(q));
scanf("%lld%lld",&n,&m);
rep(i,1,n) scanf("%lld",&c[i]);
rep(i,1,n) scanf("%lld",&v[i]),sum[i]=sum[i-1]+v[i];
rep(i,1,n)
{
dp[i]=max(q[c[i]]+sum[i],dp[i-1]);
q[c[i]]=max(q[c[i]],dp[i-1]-sum[i-1]);
ans=max(ans,dp[i]);
}
rep(i,1,m) ans=max(ans,q[i]);
printf("%lld",ans);
}