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。
题解:
如果我们事先预处理出所有
(1,i)
区间的答案并存在a[i]中的话,对于左端点在1的情况,我们只需要一个线段树查询所有的a[i]的最大值就好了。那么对于每一个左端点我们如何查询答案呢?
假设我们知道了以
i
为左端点的答案所构建的线段树的话,那么我们知道以
当然对于左端点为
i
的情况来说,可能答案区间
最后祝您,注意long long
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<algorithm>
#define ll long long
#define LiangJiaJun main
using namespace std;
int f[1000004],n,m,w[1000004];
struct segment{
int l,r;ll w,tag;
}tr[4000004];
int nt[1000004],bf[1000004];
ll pre[1000004];
int vis[1000004];
void pushdown(int k){
ll x=tr[k].tag;
tr[k].tag=0;
tr[k<<1].w+=x;tr[k<<1].tag+=x;
tr[k<<1|1].w+=x;tr[k<<1|1].tag+=x;
}
void build(int k,int l,int r){
tr[k].l=l;tr[k].r=r;
if(l==r)return;
int mid=(l+r)>>1;
build(k<<1,l,mid);build(k<<1|1,mid+1,r);
return;
}
void modify(int k,int a,int b,ll x){
if(b<a)return;
if(tr[k].tag&&tr[k].l!=tr[k].r)pushdown(k);
int l=tr[k].l,r=tr[k].r;
if(l==a&&r==b){
tr[k].w+=x;
tr[k].tag+=x;
return;
}
int mid=(l+r)>>1;
if(b<=mid)modify(k<<1,a,b,x);
else if(a>mid)modify(k<<1|1,a,b,x);
else{
modify(k<<1,a,mid,x);
modify(k<<1|1,mid+1,b,x);
}
tr[k].w=max(tr[k<<1].w,tr[k<<1|1].w);
return;
}
int LiangJiaJun(){
memset(vis,0,sizeof(vis));
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)scanf("%d",&f[i]);
for(int i=1;i<=m;i++)scanf("%d",&w[i]);
for(int i=n;i>=1;i--){
nt[i]=bf[f[i]];
bf[f[i]]=i;
}
build(1,1,n);
for(int i=1;i<=m;i++)if(bf[i])modify(1,bf[i],nt[bf[i]]-1>0?nt[bf[i]]-1:n,w[i]);
ll ans=0;
for(int i=1;i<=n;i++){
ans=max(ans,tr[1].w);
if(nt[i]){
modify(1,i+1,nt[i]-1,-w[f[i]]);
modify(1,nt[i],nt[nt[i]]-1>0?nt[nt[i]]-1:n,w[f[i]]);
}else{
modify(1,i+1,n,-w[f[i]]);
}
}
printf("%lld\n",ans);
return 0;
}