借助 dp 公式去优化

题目描述

一天，神犇和 LCR 在玩扑克牌。他们玩的是一种叫作“接竹竿”的游戏。

游戏规则是：一共有 nnn 张牌，每张牌上有一个花色 ccc 和一个点数 vvv，花色不超过 kkk 种。将这些牌依次放入一列牌的末端。若放入以前这列牌中已有与这张牌花色相同的牌，你能够选择将这张牌和任意一张花色相同的牌之间的全部牌所有取出队列（包括这两张牌自己），并获得与取出的全部牌点数和相同的分数。如今已知 LCR 把这 nnn 张牌放入队列的顺序，求她最多能得多少分。

输入顺序即为 LCR 放入队列的顺序。即，cic_ic​i​​ 表示第 iii 张放入的牌的花色，viv_iv​i​​ 表示第 iii 张放入的牌的点数。

请注意，若是你知道相似的纸牌游戏，请尤为仔细地阅读规则，以避免由于理解题意错误而出现没必要要的问题。

输入格式

第一行两个整数 n,kn,kn,k。
第二行，nnn 个整数 c1,c2,...,cnc_1,c_2,...,c_nc​1​​,c​2​​,...,c​n​​ 表示花色，知足 1≤ci≤k1\leq c_i\leq k1≤c​i​​≤k。
第三行，nnn 个整数 v1,v2,...,vnv_1,v_2,...,v_nv​1​​,v​2​​,...,v​n​​ 表示点数。

输出格式

输出一行一个整数，表示最多能获得的分数。

样例

样例输入 1

7 3
1 2 1 2 3 2 3
1 2 1 2 3 2 3

样例输出 1

13

样例解释 1

第 1 步，向队列加入 111。如今的队列：111
第 2 步，向队列加入 222。如今的队列：1,21,21,2
第 3 步，向队列加入 111。如今的队列：1,2,11,2,11,2,1
第 4 步，向队列加入 222，取出 2,1,22,1,22,1,2。如今的队列：111
第 5 步，向队列加入 333。如今的队列：1,31,31,3
第 6 步，向队列加入 222。如今的队列：1,3,21,3,21,3,2
第 7 步，向队列加入 333，取出 3,2,33,2,33,2,3。如今的队列：111

样例输入 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

数据范围与提示

对于 100%100\%100% 的数据，1≤n≤106,1≤k≤106,1≤vi≤1091\leq n\leq 10^6,1\leq k\leq 10^6,1\leq v_i\leq 10^91≤n≤10​

6​​,1≤k≤10​6​​,1≤v​i​​≤10​9​​。

题意 ：相似于“拉火车”游戏，不过不太同样的是当出现相同的状况后你能够选择拉或者不拉，

思路分析 ： 一看就是个 dp ,

　　　　　　dp[i] = dp[i-1]

　　　　　　dp[i] = max(dp[i], dp[j-1]+sum[i]-sum[j-1]) , i 与 j 为花色相同的地方

　　　　　　第一次写的时候就相似纯暴力的了，每次转移的时候从前面全部相同的地方转移，用 vector 存的，超时.... 后面发现 转移方程中的式子是能够优化的，每次找 dp[j-1]-sum[j-1]中最大的转移便可

代码示例 ：

#define ll long long
const ll maxn = 1e6+5;

ll n, k;
ll c[maxn], v[maxn];
ll sum[maxn], dp[maxn];
ll color[maxn];

void solve() {
    memset(color, 0x8f, sizeof(color)); 
    for(ll i = 1; i <= n; i++){
        dp[i] = dp[i-1];
        dp[i] = max(dp[i], sum[i]+color[c[i]]);
        color[c[i]] = max(color[c[i]], dp[i-1]-sum[i-1]); 
    }
    printf("%lld\n", dp[n]);
}

int main() {
    //freopen("in.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
    
    cin >> n >> k;
    for(ll i = 1; i <= n; i++) {
        scanf("%lld", &c[i]);
    } 
    for(ll i = 1; i <= n; i++) {
        scanf("%lld", &v[i]);
        sum[i] = sum[i-1]+v[i];
    }
    
    solve();
    return 0;
}