题解:本题主要考查线段树+枚举
简要题意:m部电影,好看值为w[i],n天,每天放f[i]部。观看第l,l+1,…,r天内所有的电影,如果同部电影观看多于一次就无法获得这部电影的好看值,求最大化观看且仅观看过一次的电影好看值总和。
1.线段树:求最大子段和,同时看过的我们可以让它为负数,抵消好看值。线段树维护
2.枚举:枚举右端点,pre[i] + 1到i段加,重复出现就pre[pre[i]] + 1到pre[i]段减,查询[1, i]的最大值(注意开long long)
代码如下:
#include<iostream>
#include<algorithm>
using namespace std;
struct N
{
long long ans,lai;
}t[2938924];
long long n,m,ans;
long long a[2483694],b[2325289],last[2293274],pre[2323815];//pre[i]表示与i这个位置上的数相等的前一个数的下标
void pushup(int p)
{
t[p].ans=max(t[p*2].ans,t[p*2+1].ans);
}
void lazy(int p)
{
if(t[p].lai)
{
t[p*2].lai+=t[p].lai;
t[p*2+1].lai+=t[p].lai;
t[p*2].ans+=t[p].lai;
t[p*2+1].ans+=t[p].lai;
t[p].lai=0;
}
return ;
}
void change(int p,int l,int r,int x,int y,int z)
{
if(l>=x&&r<=y)
{
t[p].ans+=z;
t[p].lai+=z;
return ;
}
lazy(p);
int mid=l+r>>1;
if(x<=mid)change(p*2,l,mid,x,y,z);
if(y>mid)change(p*2+1,mid+1,r,x,y,z);
pushup(p);
}
long long ask(int p,int l,int r,int x,int y)
{
if(l>=x&&r<=y)return t[p].ans;
lazy(p);
int mid=l+r>>1;
long long sum=0;
if(x<=mid)sum=max(sum,ask(p*2,l,mid,x,y));
if(y>mid)sum=max(sum,ask(p*2+1,mid+1,r,x,y));
return sum;
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)cin>>a[i];
for(int i=1;i<=m;i++)cin>>b[i];
for(int i=1;i<=n;i++){pre[i]=last[a[i]];last[a[i]]=i;}
for(int i=1;i<=n;i++)
{
change(1,1,n,pre[i]+1,i,b[a[i]]);
if(pre[i])change(1,1,n,pre[pre[i]]+1,pre[i],-b[a[i]]);
ans=max(ans,ask(1,1,n,1,i));
}
cout<<ans;
return 0;
}