【bzoj3747】[POI2015]Kinoman

时间:2022-02-11 21:54:17

题解:

水题

从左向右维护以每一个作为右端点的最大值

线段树维护

代码:

#include <bits/stdc++.h>
using namespace std;
#define rint register ll
#define IL inline
#define rep(i,h,t) for (rint i=h;i<=t;i++)
#define dep(i,t,h) for (rint i=t;i>=h;i--)
#define mid ((h+t)>>1)
#define ll long long
const ll N=1e6+;
const ll N2=N*;
ll a[N],b[N],p1[N],p2[N];
struct sgt{
ll maxa[N2],lazy[N2];
IL void down(rint x)
{
maxa[x*]+=lazy[x]; maxa[x*+]+=lazy[x];
lazy[x*]+=lazy[x]; lazy[x*+]+=lazy[x];
lazy[x]=;
}
#define updata(x) maxa[x]=max(maxa[x*2],maxa[x*2+1])
void change(ll x,ll h,ll t,ll h1,ll t1,ll k)
{
if (h1<=h&&t<=t1)
{
lazy[x]+=k; maxa[x]+=k; return;
}
down(x);
if (h1<=mid) change(x*,h,mid,h1,t1,k);
if (mid<t1) change(x*+,mid+,t,h1,t1,k);
updata(x);
}
}S;
int main()
{
freopen("1.in","r",stdin);
freopen("1.out","w",stdout);
ios::sync_with_stdio(false);
ll n,m;
cin>>n>>m;
rep(i,,n) cin>>a[i];
rep(i,,m) cin>>b[i];
ll ans=;
rep(i,,n)
{
ll x=a[i];
S.change(,,n,p1[x]+,i,b[a[i]]);
if (p2[x]+<=p1[x]) S.change(,,n,p2[x]+,p1[x],-b[a[i]]);
ans=max(ans,S.maxa[]);
p2[a[i]]=p1[a[i]];
p1[a[i]]=i;
}
cout<<ans<<endl;
return ;
}