BZOJ 3747 Kinoman看电影 - 线段树

线段树写起来好容易挂，还是太渣了。。。

一般像这种统计次数的，只能出现一种，要联想到可以转化为pre数组或者next数组在[l,r]的范围之外的有多少个数。

对于这道题还需要加点边界，不然很容易就爆了。。。

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<algorithm>

using namespace std;

const int maxn=1000005;

struct tree
{
    long long add,maxi;
}t[maxn<<2];

int n,m;
int last[maxn],next[maxn];
long long val[maxn],s[maxn];
bool vst[maxn];

void maintain(int ro)
{
    t[ro].maxi=max(t[ro<<1].maxi,t[ro<<1|1].maxi);
}
void pushdown(int ro)
{
    if(t[ro].add)
    {
        t[ro<<1].add+=t[ro].add;
        t[ro<<1|1].add+=t[ro].add;
        t[ro<<1].maxi+=t[ro].add;
        t[ro<<1|1].maxi+=t[ro].add;
        t[ro].add=0;
    }
}
void addition(int ro,int L,int R,long long add,int l,int r)
{
    if(l==L&&r==R)
    {
        t[ro].add+=add;
        t[ro].maxi+=add;
        return;
    }
    pushdown(ro);
    int mid=l+r>>1;
    if(R<=mid)addition(ro<<1,L,R,add,l,mid);
    else if(L>=mid+1)addition(ro<<1|1,L,R,add,mid+1,r);
    else addition(ro<<1,L,mid,add,l,mid),addition(ro<<1|1,mid+1,R,add,mid+1,r);
    maintain(ro);
}
long long query(int ro,int L,int R,int l,int r)
{
    if(l==L&&r==R)
        return t[ro].maxi;
    pushdown(ro);
    int mid=l+r>>1;
    if(R<=mid)return query(ro<<1,L,R,l,mid);
    else if(L>=mid+1)return query(ro<<1|1,L,R,mid+1,r);
    else return max(query(ro<<1,L,mid,l,mid),query(ro<<1|1,mid+1,R,mid+1,r));
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",s+i);
        next[last[s[i]]]=i;
        last[s[i]]=i;
    }
    for(int i=1;i<=m;i++)if(last[i])
        next[last[i]]=n+1;
    for(int i=1;i<=m;i++)
        scanf("%d",val+i);
    for(int i=1;i<=n;i++)
        if(!vst[s[i]])addition(1,i,next[i]-1,val[s[i]],1,n),vst[s[i]]=true;
    long long ans=0;
    for(int i=1;i<n;i++)
    {
        ans=max(ans,query(1,i,n,1,n));
        addition(1,i,next[i]-1,-val[s[i]],1,n);
        if(next[next[i]])addition(1,next[i],next[next[i]]-1,val[s[i]],1,n);
    }
    ans=max(ans,query(1,n,n,1,n));
    printf("%lld",ans);
    return 0;
}