[POI2015]Kinoman

题目大意：
　　给你一个长度为n的数列f，f中共有m种不同的数，每种数都有一个权值w[i]。
　　你可以选定一个f中的区间，定义区间的权值为这一区间只出现一次的数的权值和。
　　问权值最大的区间的权值是多少？

思路：
　　对于f中的每一个位置i，找到下一个和它相同数字的位置next[i]。
　　从左到右枚举区间左端点，线段树维护选取每个右端点的最大值。
　　去除当前左端点对答案的影响时，只需要把i~next[i]-1这一段减去w[f[i]]，然后把next[i]~next[next[i]]-1加上w[f[i]]即可。

#include<cstdio>
#include<cctype>
#include<algorithm>
typedef long long int64;
inline int getint() {
    register char ch;
    while(!isdigit(ch=getchar()));
    register int x=ch^'0';
    while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
    return x;
}
const int N=1000001,M=1000001;
int f[N],w[M],next[N],pos[M];
class SegmentTree {
    #define _left <<1
    #define _right <<1|1
    private:
        int64 max[N<<2],tag[N<<2];
        void push_down(const int &p) {
            tag[p _left]+=tag[p];
            tag[p _right]+=tag[p];
            max[p _left]+=tag[p];
            max[p _right]+=tag[p];
            tag[p]=0;
        }
        void push_up(const int &p) {
            max[p]=std::max(max[p _left],max[p _right]);
        }
    public:
        void modify(const int &p,const int &b,const int &e,const int &l,const int &r,const int &x) {
            if(b==l&&e==r) {
                tag[p]+=x;
                max[p]+=x;
                return;
            }
            push_down(p);
            const int mid=(b+e)>>1;
            if(l<=mid) modify(p _left,b,mid,l,std::min(mid,r),x);
            if(r>mid) modify(p _right,mid+1,e,std::max(mid+1,l),r,x);
            push_up(p);
        }
        int64 query() const {
            return max[1];
        }
    #undef _left
    #undef _right
};
SegmentTree t;
int main() {
    const int n=getint(),m=getint();
    for(register int i=1;i<=n;i++) {
        f[i]=getint();
    }
    for(register int i=1;i<=m;i++) {
        w[i]=getint();
        pos[i]=n+1;
    }
    for(register int i=n;i;i--) {
        next[i]=pos[f[i]];
        pos[f[i]]=i;
    }
    for(register int i=1;i<=m;i++) {
        t.modify(1,1,n,pos[i],next[pos[i]]-1,w[i]);
    }
    int64 ans=0;
    for(register int i=1;i<=n;i++) {
        ans=std::max(ans,t.query());
        t.modify(1,1,n,i,next[i]-1,-w[f[i]]);
        t.modify(1,1,n,next[i],next[next[i]]-1,w[f[i]]);
    }
    printf("%lld\n",ans);
    return 0;
}