[BZOJ3747]-[POI2015]Kinoman-线段树套路题

说在前面

刷水题
me居然没有1A，不开心

---

题目

BZOJ3747传送门

题面

一共有 $m$ 部电影，编号为1~m，第 $i$ 部电影的好看值为 $w_i$
在 $n$ 天之中（从1~n编号）每天会放映一部电影，第 $i$ 天放映的是第 $f_i$ 部。
你可以选择$l,r$ $(1\leq l\leq r\leq n)$，并观看第$l,l+1,\cdots ,r$天内所有的电影。如果同一部电影你观看多于一次，你会感到无聊，于是无法获得这部电影的好看值。所以你希望最大化观看且仅观看过一次的电影的好看值的总和
范围：$n,m,w_i\leq 1000000$，$f_i\leq m$

输入输出格式

输入格式：
第一行两个整数$n,m$，含义如题
接下来一行 $n$ 个整数，描述 $f_1$ 到 $f_n$
接下来一行 $m$ 个整数，描述 $w_1$ 到 $w_n$

输出格式：
输出一行一个整数表示答案

---

解法

像这种题，扫一扫右端点，然后线段树维护当前右端点时，左端点选 i i <script type="math/tex" id="MathJax-Element-480">i</script>的贡献是多少就好了
反正相同套路的题很多

---

下面是代码

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std ;

int N , M , mv[1000005] , hp[1000005] , head[1000005] , pre[1000005] ;
struct Node{
    long long mx , flag ;
    Node *ch[2] ;
    void Add( long long x ){ flag += x , mx += x ; }
    void pushdown(){
        if( !flag ) return ; 
        ch[0]->Add( flag ) , ch[1]->Add( flag ) ;
        flag = 0 ;
    }
    void update(){ mx = max( ch[0]->mx , ch[1]->mx ) ; }
} w[2000005] , *tw = w , *root ;

template<typename T>
void smax( T &x , T y ){ if( x < y ) x = y ; }

Node *build( int lf , int rg ){
    Node *nd = ++tw ;
    if( lf != rg ){
        int mid = ( lf + rg ) >> 1 ;
        nd->ch[0] = build( lf , mid ) ;
        nd->ch[1] = build( mid+1,rg ) ;
    } return nd ;
}

void Modify( Node *nd , int lf , int rg , int L , int R , int delta ){
    if( L <= lf && rg <= R ){ nd->Add( delta ) ; return ; }
    int mid = ( lf + rg ) >> 1 ; nd->pushdown() ;
    if( L <= mid ) Modify( nd->ch[0] , lf , mid , L , R , delta ) ;
    if( R >  mid ) Modify( nd->ch[1] , mid+1,rg , L , R , delta ) ;
    nd->update() ;
}

void solve(){
    long long ans = 0 ;
    for( int i = 1 ; i <= N ; i ++ ){
        if( pre[i] ) Modify( root , 1 , N , pre[ pre[i] ] + 1 , pre[i] , -hp[ mv[i] ] ) ;
        Modify( root , 1 , N , pre[i] + 1 , i , hp[ mv[i] ] ) ;
        smax( ans , root->mx ) ;
    } printf( "%lld\n" , ans ) ;
}

void read_( int &x ){
    register char ch = getchar() ;
    while( ch < '0' || ch > '9' ) ch = getchar() ;
    while( ch >='0' && ch <='9' ) x = ( x << 1 ) + ( x << 3 ) + ch - '0' , ch = getchar() ;
}

int main(){
    scanf( "%d%d" , &N , &M ) ;
    for( int i = 1 ; i <= N ; i ++ ){
        read_( mv[i] ) ;
        pre[i] = head[ mv[i] ] , head[ mv[i] ] = i ;
    } for( int i = 1 ; i <= M ; i ++ ) read_( hp[i] ) ;
    root = build( 1 , N ) ; solve() ;
}