Wannafly挑战赛21 C 大水题

题目描述
现在给你N个正整数ai，每个数给出一“好数程度” gi（数值相同但位置不同的数之间可能有不同的好数程度）。对于在 i 位置的数，如果有一在j位置的数满足 j < i 且 ai=aj，则你可以将位于[i,j]闭区间内的序列评为“好序列”，然后获得∑gk(j≤k≤i)（此闭区间内“好数程度”之和）分数。
注意: 在所有情况下，每个数都只能被一个”好序列”包含（只能与其他相应数被评为”好序列”一次）；在符合要求的情况下，”好序列”的评定次数不受限制，且通过不同”好序列”获得的分数可以累加。

输入描述:
第一行有一个正整数N。
接下来的一行有N个正整数ai，表示意义如上。
（保证ai在32位整型范围内）
接下来的一行有N个正整数gi，表示ai的”好数程度”。
（保证gi在64位整型范围内）
输出描述:
一个整数，你可以获得的最大分数（通过不同”好序列”获得的分数可以累加），保证答案在64位整型范围内。
示例1
输入
复制
7
1 2 1 2 3 2 3
1 4 3 4 3 4 5
输出
复制
23
备注:
数据范围 2≤N≤300000

本来是用next做的，可是那样很麻烦，应该用pre做，这样可以少考虑一些情况而且代码会比较简单，
$dp[i][j]，i表示的是位置，j表示的是是否取这位数$
所有的数只能取一遍，所以在dp[vis[i]][1]的时候不用减一

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=3e5+5;
ll a[maxn],b[maxn];
ll dp[maxn][2];
map<ll,int>vis;
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%lld",&a[i]);
    for(int i=1;i<=n;i++)
        scanf("%lld",&b[i]),b[i]+=b[i-1];
    for(int i=1;i<=n;i++)
    {
        dp[i][0]=max(dp[i-1][0],dp[i-1][1]);
        if(vis[a[i]])
        {
            dp[i][1]=max(dp[i][1],dp[vis[a[i]]][0]+b[i]-b[vis[a[i]]-1]);
            dp[i][1]=max(dp[i][1],dp[vis[a[i]]][1]+b[i]-b[vis[a[i]]]);
        }
        vis[a[i]]=i;
    }
    printf("%lld\n",max(dp[n][1],dp[n][0]));
    return 0;
}