共有m部电影,编号为1~m,第i部电影的好看值为w[i]。
在n天之中(从1~n编号)每天会放映一部电影,第i天放映的是第f[i]部。
你可以选择l,r(1<=l<=r<=n),并观看第l,l+1,…,r天内所有的电影。如果同一部电影你观看多于一次,你会感到无聊,于是无法获得这部电影的好看值。所以你希望最大化观看且仅观看过一次的电影的好看值的总和。
第一行两个整数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)。
输出观看且仅观看过一次的电影的好看值的总和的最大值。
这里考虑每一个电影对某一个区间的影响,用线段树维护一个区间最大值
记为第个电影前一次出现的位置,为第个电影的编号,那么就是第个电影的好看值
那么第个电影对这个区间显然有的贡献,那么用线段树区间更新,
同时第个电影,对之前更新过的区间有着一个的贡献
(即消除之前的影响,因为每次查询区间最大值查询的是,即中最大的区间最大值
,因为中所有的电影都处理过了,所以可以查询)
注意这里当之前更新过(不为0)的时候才能对之前的区间进行取消更新操作,
否则就相当于对()进行了取消更新操作,这就错了
还有,注意懒标记要开longlong,因为范围超过int了
#include
using namespace std;
typedef long long ll;
const int maxn = 1e6+7;
ll a[maxn],n,m,pre[maxn],w[maxn],last[maxn];
struct node{
int l,r;
ll Maxsum,lazy;
void updata(ll val){
Maxsum += val;
lazy += val;
}
}tree[maxn<<2];
void push_up(int x){
tree[x].Maxsum = max(tree[x<<1].Maxsum, tree[x<<1|1].Maxsum);
}
void push_down(int x){
ll lazy = tree[x].lazy;
if(lazy){
tree[x].lazy = 0;
tree[x<<1].updata(lazy);
tree[x<<1|1].updata(lazy);
}
}
void build(int x,int l,int r){
tree[x].l = l,tree[x].r = r;
tree[x].lazy = tree[x].Maxsum = 0;
if(l != r){
int mid = (l+r)>>1;
build(x<<1,l,mid);
build(x<<1|1,mid+1,r);
push_up(x);
}
}
void updata(int x,int l,int r,ll val){
int L = tree[x].l,R = tree[x].r;
if(l<=L && R<=r)
tree[x].updata(val);
else{
int mid = (L+R)>>1;
push_down(x);
if(l<=mid) updata(x<<1,l,r,val);
if(r>mid) updata(x<<1|1,l,r,val);
push_up(x);
}
}
ll query(int x,int l,int r){
int L = tree[x].l,R = tree[x].r;
if(l<=L && R<=r)
return tree[x].Maxsum;
else{
int mid = (L+R)>>1;
push_down(x);
ll ans = 0;
if(l<=mid) ans = max(ans, query(x<<1,l,r));
if(r>mid) ans = max(ans, query(x<<1|1,l,r));
push_up(x);
return ans;
}
}
int main(){
scanf("%lld%lld",&n,&m);
for(int i=1;i<=n;i++)
scanf("%lld",&a[i]);
for(int i=1;i<=m;i++)
scanf("%lld",&w[i]);
for(int i=1;i<=n;i++){
pre[i] = last[a[i]];
last[a[i]] = i;
}
build(1,1,n);
ll ans = 0;
for(int i=1;i<=n;i++){
updata(1,pre[i]+1,i,w[a[i]]);
if(pre[i]) updata(1,pre[pre[i]]+1,pre[i],-w[a[i]]);
ans = max(ans,query(1,1,i));
}
printf("%lld\n",ans);
return 0;
}