3747: [POI2015]Kinoman

3747: [POI2015]Kinoman

Time Limit: 60 Sec   Memory Limit: 128 MB
Submit: 783   Solved: 318
[ Submit][ Status][ Discuss]

Description

共有m部电影，编号为1~m，第i部电影的好看值为w[i]。

在n天之中（从1~n编号）每天会放映一部电影，第i天放映的是第f[i]部。

你可以选择l,r(1<=l<=r<=n)，并观看第l,l+1,…,r天内所有的电影。如果同一部电影你观看多于一次，你会感到无聊，于是无法获得这部电影的好看值。所以你希望最大化观看且仅观看过一次的电影的好看值的总和。

Input

第一行两个整数n,m(1<=m<=n<=1000000)。

第二行包含n个整数f[1],f[2],…,f[n](1<=f[i]<=m)。

第三行包含m个整数w[1],w[2],…,w[m](1<=w[j]<=1000000)。

Output

输出观看且仅观看过一次的电影的好看值的总和的最大值。

Sample Input

9 4
2 3 1 1 4 1 2 4 1
5 3 6 6

Sample Output

15
样例解释：
观看第2,3,4,5,6,7天内放映的电影，其中看且仅看过一次的电影的编号为2,3,4。

HINT

Source

鸣谢Jcvb

[ Submit][ Status][ Discuss]



任意选择一段区间，让它右端点的电影一定能获得好看值，这样的决策肯定不会差

如果右端点的电影无法获得好看值，去掉它，前面的一部电影可能有贡献，就算没有，只是相当于-0

贪心地想，枚举每个右端点，为每个右端点选择一个合适的左端点

假设当前处理的右端点为R，最优左端点为L，不妨为[L,R]里的数字赋予权值

对于任意数字i，从R往L，第一次出现时赋w[i]，第二次出现-w[i]，三次及以上为0

这样[L,R]的权值和就是观赏[L,R]的电影能获得的好看值了

如果从左往右处理R，那上面的东西就相当于一个个后缀和

单点修改 = 区间修改后缀和，线段树维护就行了  O(nlogn)

Modify老是忘记打pushdown..........

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#include<algorithm>
#include<cmath>
using namespace std;

const int maxn = 1E6 + 10;
typedef long long LL;
const int T = 4;

int n,m,f[maxn],w[maxn],pos[maxn][2];
LL Ans,Max[maxn*T],Add[maxn*T];

void pushdown(int o,int l,int r)
{
	if (!Add[o]) return; Max[o] += Add[o];
	if (l == r) {Add[o] = 0; return;}
	Add[o<<1] += Add[o]; Add[o<<1|1] += Add[o]; Add[o] = 0;
}

void Modify(int o,int l,int r,int ml,int mr,LL k)
{
	if (ml <= l && r <= mr)
	{
		Add[o] += k;
		pushdown(o,l,r);
		return;
	}
	int mid = (l + r) >> 1; pushdown(o,l,r);
	if (ml <= mid) Modify(o<<1,l,mid,ml,mr,k); else pushdown(o<<1,l,mid);
	if (mr > mid) Modify(o<<1|1,mid+1,r,ml,mr,k); else pushdown(o<<1|1,mid+1,r);
	Max[o] = max(Max[o<<1],Max[o<<1|1]);
}

LL Query(int o,int l,int r,int ql,int qr)
{
	pushdown(o,l,r);
	if (ql <= l && r <= qr) return Max[o];
	int mid = (l + r) >> 1; LL ret = -1E16;
	if (ql <= mid) ret = Query(o<<1,l,mid,ql,qr);
	if (qr > mid) ret = max(ret,Query(o<<1|1,mid+1,r,ql,qr));
	return ret;
}

void Insert(int num,int po)
{
	if (!pos[num][0])
	{
		pos[num][0] = po;
		Modify(1,1,n,1,po,w[num]);
	}
	else if (!pos[num][1])
	{
		pos[num][1] = po;
		Modify(1,1,n,1,pos[num][0],-2*w[num]);
		Modify(1,1,n,1,po,w[num]);
	}
	else
	{
		Modify(1,1,n,1,pos[num][0],w[num]);
		pos[num][0] = pos[num][1];
		Modify(1,1,n,1,pos[num][0],-2*w[num]);
		pos[num][1] = po;
		Modify(1,1,n,1,po,w[num]);
	}
}

int getint()
{
	char ch = getchar(); int ret = 0;
	while (ch < '0' || '9' < ch) ch = getchar();
	while ('0' <= ch && ch <= '9')
		ret = ret*10 + ch - '0',ch = getchar();
	return ret;
}

int main()
{
	#ifdef DMC
		freopen("DMC.txt","r",stdin);
	#endif
	
	n = getint(); m = getint();
	for (int i = 1; i <= n; i++) f[i] = getint();
	for (int i = 1; i <= m; i++) w[i] = getint();
	for (int i = 1; i <= n; i++)
		Insert(f[i],i),Ans = max(Ans,Query(1,1,n,1,i));
	cout << Ans;
	return 0;
}