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

区间dp,但是常规区间dp方程式是枚举所有子区间

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
            }
        }
    }

dp[i]=max(dp[j-1]+sum(j~i))a[j]==a[i];
会超时，记录相同数字的最优点
详细看代码：

#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;
}