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

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstdlib>
#include <cstring>
#include <string>
#include <deque>
#include <map>
using namespace std;
#define ll long long 
#define  N 300008
#define  gep(i,a,b)  for(int  i=a;i<=b;i++)
#define  gepp(i,a,b) for(int  i=a;i>=b;i--)
#define  gep1(i,a,b)  for(ll i=a;i<=b;i++)
#define  gepp1(i,a,b) for(ll i=a;i>=b;i--)    
#define  mem(a,b)  memset(a,b,sizeof(a))
ll sum[N],g[N],dp[N];
int a[N],pre[N];
int n;
map<int,int>mp;
int main()
{
    scanf("%d",&n);
    gep(i,1,n) 
    {
        scanf("%d",&a[i]);
        pre[i]=mp[a[i]];//记录前面a[i]==a[j]的j的位置
        mp[a[i]]=i;
    }
    gep(i,1,n){
        scanf("%lld",&g[i]);
        sum[i]=sum[i-1]+g[i];//前缀和
    }
    int j;
    gep(i,2,n){
        dp[i]=dp[i-1];//初始化为dp[i-1]
        j=pre[i];
        while(j){
            dp[i]=max(dp[i],sum[i]-sum[j-1]+dp[j-1]);//每个数只能用一次
            if(dp[i]==sum[i])  break;//一定最大了，不然会超时
            j=pre[j];//前面的j都是可能的
        }
    }
    printf("%lld\n",dp[n]);
    return 0;
}