Wannafly挑战赛21 C 大水题

题目链接：​​点我​​​ 链接：https://www.nowcoder.com/acm/contest/159/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

for (int len = 1; len < n; len++) { //操作区间的长度
        for (int i = 0, j = len; j <= n; i++, j++) { //始末
            //检查是否匹配（非必须）
            for (int s = i; s < j; s++) {
                //update
            }
        }
    }      

#include<iostream>
#include<map>
#include<algorithm>
const int maxn=300000+10;
typedef long long ll;
using namespace std;
map<int,int> p;
int a[maxn],best[maxn],last[maxn];
ll dp[maxn],sum[maxn];
int main(){
  int n;
  scanf("%d",&n);
  for(int i=1;i<=n;i++){
    scanf("%d",&a[i]);
    if(p.count(a[i])==0){
      p[a[i]]=i;
      
    }else{
      last[i]=p[a[i]];
      p[a[i]]=i;
    }
  }
  
  int lin;
  
  for(int i=1;i<=n;i++)
  {
    scanf("%lld", &sum[i]);
    sum[i] += sum[i-1];
  }
  ll ans=0;
  for(int i=1;i<=n;i++){
    if(last[i]!=0){ 
      dp[i]=dp[last[i]-1]+sum[i]-sum[last[i]-1];
      best[i]=last[i]-1;
      if(best[last[i]]!=0&&dp[best[last[i]]]+sum[i]-sum[best[last[i]]]>dp[i]){
        dp[i]=dp[best[last[i]]]+sum[i]-sum[best[last[i]]];
        best[i] = best[last[i]];
      }  //比如数据里有3个2，就会比较前面那个2好一点，等到下次再遇到2时，再比较，但是
      //每次只用比较前面那个2和最好的那个2就行
    }
    ans=max(ans,dp[i]);
  }
  printf("%lld\n",ans);
  return 0;
}