题目大意:给定f[],g[]数组,在f[]数组中选择一个区间[l,r],要求最大化 fl,r 中只出现一次的数的权值
题解:经典套路
枚举区间左端点,线段树维护每个位置作为右端点的答案,取max
首先预处理出l=1的情况,然后每次l+1时,
[l,next[l]−1]
的答案减少
w[f[l]]
[next[l],next[next[l]]−1]
的答案增加
w[f[l]]
r<l 的答案不需要特殊处理
我的收获:套路++
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define INF 0x3f3f3f3f
#define M 1000005
#define ls x<<1
#define rs x<<1|1
#define lson l,mid,x<<1
#define rson mid+1,r,x<<1|1
#define root 1,n,1
int n,m,f[M],w[M];
int next[M],last[M];
ll add[M<<2],mx[M<<2],ans;
bool mark[M];
void pushup(int x){mx[x]=max(mx[ls],mx[rs]);}
void pushdown(int x,int m)
{
if(!add[x]) return ;
add[ls]+=add[x],add[rs]+=add[x];
mx[ls]+=add[x],mx[rs]+=add[x];
add[x]=0;
}
void updata(int L,int R,int v,int l,int r,int x)
{
if(L<=l&&r<=R){add[x]+=v,mx[x]+=v;return ;}
pushdown(x,r-l+1);
int mid=(l+r)>>1;
if(L<=mid) updata(L,R,v,lson);
if(R>mid) updata(L,R,v,rson);
pushup(x);
}
void work()
{
for(int i=1;i<=n;i++)
{
ans=max(ans,mx[1]);
int t=next[i];
if(t){
updata(i,t-1,-w[f[i]],root);
updata(t,!next[t]?n:next[t]-1,w[f[i]],root);
}
else updata(i,n,-w[f[i]],root);
}
printf("%lld\n",ans);
}
void init()
{
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--) next[i]=last[f[i]],last[f[i]]=i;
for(int i=1;i<=m;i++) if(last[i]) updata(last[i],!next[last[i]]?n:next[last[i]]-1,w[i],root);
}
int main()
{
init();
work();
return 0;
}