共有m部电影,编号为1~m,第i部电影的好看值为w[i]。 在n天之中(从1~n编号)每天会放映一部电影,第i天放映的是第f[i]部。
你可以选择l,r(1<=l<=r<=n),并观看第l,l+1,…,r天内所有的电影。如果同一部电影你观看多于一次,你会感到无聊,于是无法获得这部电影的好看值。所以你希望最大化观看且仅观看过一次的电影的好看值的总和。
这道题算挺好的吧,也不是特别难想的。看这种题就应该想到数据结构(线段树),但要维护什么呢?一般来说,它问什么就维护什么,所以我们就维护s[l]~s[n]的最大值(l为当前的左端点,s[i]表示当前的左端点到i这部分电影的总好看值),感觉这道题跟采花这道题比较像。
我们可以靠推进左端点,来计算全部的区间,但推进的时候,对s数组也会有改动,那我们需要用next数组记录下一个离它最近的跟它相同的电影的位置,那么就很好维护了,区间修改的经典思想。之后,这道题就解决了。
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
struct trnode
{
int l,r,lc,rc;long long c,lazy;
trnode(){lazy=0;}
}tr[2100000];int trlen;
int a[1100000],b[1100000],num[1100000],next[1100000],wy[1100000];
long long s[1100000];
void bt(int l,int r)
{
trlen++;int now=trlen;
tr[now].l=l;tr[now].r=r;
tr[now].lc=tr[now].rc=-1;
if(l==r)tr[now].c=s[l];
else
{
int mid=(l+r)/2;
tr[now].lc=trlen+1;bt(l,mid);
tr[now].rc=trlen+1;bt(mid+1,r);
tr[now].c=max(tr[tr[now].lc].c,tr[tr[now].rc].c);
}
}
void update(int now)
{
int lc=tr[now].lc,rc=tr[now].rc;
tr[lc].c+=tr[now].lazy;tr[lc].lazy+=tr[now].lazy;
tr[rc].c+=tr[now].lazy;tr[rc].lazy+=tr[now].lazy;
tr[now].lazy=0;
}
void change(int now,int l,int r,int k)
{
if(l>r)return ;
if(tr[now].l==l && tr[now].r==r){tr[now].c+=k;tr[now].lazy+=k;return ;}
if(tr[now].lazy!=0)update(now);
int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)/2;
if(r<=mid)change(lc,l,r,k);
else if(mid+1<=l)change(rc,l,r,k);
else change(lc,l,mid,k),change(rc,mid+1,r,k);
tr[now].c=max(tr[lc].c,tr[rc].c);
}
long long findmax(int now,int l,int r)
{
if(tr[now].l==l && tr[now].r==r)return tr[now].c;
if(tr[now].lazy!=0)update(now);
int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)/2;
if(r<=mid)return findmax(lc,l,r);
else if(mid+1<=l)return findmax(rc,l,r);
else return max(findmax(lc,l,mid),findmax(rc,mid+1,r));
}
int main()
{
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)scanf("%d",&a[i]);
for(int i=1;i<=m;i++)scanf("%d",&b[i]);
for(int i=n;i>=1;i--)
{
next[i]=wy[a[i]];
wy[a[i]]=i;
}
for(int i=1;i<=n;i++)
{
s[i]=s[i-1];
if(num[a[i]]==0)s[i]+=b[a[i]];
if(num[a[i]]==1)s[i]-=b[a[i]];
num[a[i]]++;
}
bt(1,n);
long long ans=0;
for(int i=1;i<=n;i++)
{
ans=max(ans,findmax(1,i,n));
if(next[i]!=0)
{
change(1,i+1,next[i]-1,-b[a[i]]);
if(next[next[i]]!=0)change(1,next[i],next[next[i]]-1,b[a[i]]);
else change(1,next[i],n,b[a[i]]);
}
else change(1,i+1,n,-b[a[i]]);
}
printf("%lld\n",ans);
return 0;
}