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
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。
线段树的一眼题。。
枚举右端点。然后维护每个点为左端点的ans
设pre[i]为f[i]上一次出现的位置
则每次枚举一个新的右端点就在[pre[i]+1,i]加w[f[i]]然后[pre[pre[i]]+1,pre[i]]减w[f[i]]就行了
线段树区间修改区间查询 维护区间最大值
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
using namespace std;
struct tree
{
long long m;
long long tag;
}tr[4000001];
inline void build(int p,int l,int r)
{
if(l!=r)
{
int mid=(l+r)/2;
build(p*2,l,mid);
build(p*2+1,mid+1,r);
}
}
inline void push(int p)
{
if(p>2000000)
return ;
tr[p*2].m+=tr[p].tag;
tr[p*2+1].m+=tr[p].tag;
tr[p*2].tag+=tr[p].tag;
tr[p*2+1].tag+=tr[p].tag;
tr[p].tag=0;
}
inline void up(int p)
{
tr[p].m=max(tr[p*2].m,tr[p*2+1].m);
}
inline void inc(int p,int ll,int rr,int l,int r,long long x)
{
if(l>r)
return ;
if(l<=ll&&rr<=r)
{
tr[p].m+=x;
tr[p].tag+=x;
}
else
{
push(p);
int mid=(ll+rr)/2;
if(l<=mid)
inc(p*2,ll,mid,l,r,x);
if(r>mid)
inc(p*2+1,mid+1,rr,l,r,x);
up(p);
}
}
inline long long ask(int p,int ll,int rr,int l,int r)
{
if(l<=ll&&rr<=r)
return tr[p].m;
else
{
push(p);
int mid=(ll+rr)/2;
long long ans=0;
if(l<=mid)
ans=max(ans,ask(p*2,ll,mid,l,r));
if(r>mid)
ans=max(ans,ask(p*2+1,mid+1,rr,l,r));
return ans;
}
}
int f[1000001],la[1000001],pre[1000001];
long long w[1000001];
int main()
{
// freopen("data.in","r",stdin);
// freopen("data.out","w",stdout);
int n,m;
scanf("%d%d",&n,&m);
int i;
for(i=1;i<=n;i++)
scanf("%d",&f[i]);
for(i=1;i<=m;i++)
scanf("%lld",&w[i]);
for(i=1;i<=n;i++)
{
pre[i]=la[f[i]];
la[f[i]]=i;
}
build(1,1,n);
long long ans=0;
int j;
for(i=1;i<=n;i++)
{
inc(1,1,n,pre[i]+1,i,w[f[i]]);
inc(1,1,n,pre[pre[i]]+1,pre[i],-w[f[i]]);
ans=max(ask(1,1,n,1,i),ans);
/*for(j=1;j<=n;j++)
printf("%lld ",ask(1,j,j));
printf("\n");*/
}
printf("%lld\n",ans);
return 0;
}