bzoj3747: [POI2015]Kinoman（線段樹）

題目傳送門
。

解法：
線段樹。
列舉左端點，用線段樹維護以每個點作為右端點的答案。
對於不同的左端點，右端點的答案是不一樣的。
設當前位置為i，電影為x，好看值為w，下個出現的位置為nxt。
在統計完以i為左端點的答案後。
我們需要往下列舉左端點。
這時候我們就算過了i這個位置。
要消除它對答案的影響。
那麼從以位置i到nxt-1為右端點都無法再獲得x的好看值。
那麼i到nxt-1都減去w。

當右端點大於等於nxt的時候，又可以重新獲得x的好看值。
但是看兩次就會得不到好看值。所以右端點在nxt到x再下一個出現的位置-1都要加上w。

過程用線段樹維護咯

程式碼實現：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
struct node {int lc,rc,l,r;ll lazy,c;}tr[2100000];int trlen;
void bt(int l,int r) {
    int now=++trlen;tr[now].l=l;tr[now].r=r;tr[now].lc=tr[now].rc=-1;tr[now].c=0;tr[now].lazy=0;
    if(l<r) {int mid=(l+r)/2;tr[now].lc=trlen+1;bt(l,mid);tr[now].rc=trlen+1;bt(mid+1,r);}
}
void update(int now) {
    int lc=tr[now].lc,rc=tr[now].rc;
    if(lc==-1)return ;
    tr[lc].c+=tr[now].lazy;tr[rc].c+=tr[now].lazy;
    tr[lc].lazy+=tr[now].lazy;tr[rc].lazy+=tr[now].lazy;tr[now].lazy=0;
}
void change(int now,int l,int r,ll k) {
    if(tr[now].l==l&&tr[now].r==r) {tr[now].c+=k;tr[now].lazy+=k;return ;}
    if(tr[now].lazy)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(l>mid)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);
}
int head[1100000],last[1100000],next[1100000],a[1100000];
ll s[1100000];
int main() {
    int n,m;scanf("%d%d",&n,&m);
    memset(head,0,sizeof(head));
    memset(last,0,sizeof(last));
    memset(next,0,sizeof(next));
    for(int i=1;i<=n;i++) {
        scanf("%d",&a[i]);if(head[a[i]]==0)head[a[i]]=i;
        next[last[a[i]]]=i;last[a[i]]=i;
    }
    for(int i=1;i<=m;i++)scanf("%lld",&s[i]);
    trlen=0;bt(1,n);
    for(int i=1;i<=m;i++) if(head[i]!=0) {
        if(next[head[i]]==0)change(1,head[i],n,s[i]);
        else change(1,head[i],next[head[i]]-1,s[i]);
    }
    ll ans=0;
    for(int i=1;i<=n;i++) {
        ans=max(ans,tr[1].c);
        int nxt=next[i];
        if(nxt!=0) {
            change(1,i,nxt-1,-s[a[i]]);
            if(next[nxt]==0)change(1,nxt,n,s[a[i]]);
            else change(1,nxt,next[nxt]-1,s[a[i]]);
        }else change(1,i,n,-s[a[i]]);
    }printf("%lld\n",ans);
    return 0;
}