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

说在前面

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

---

题目

BZOJ3747传送门

题面

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

输入输出格式

输入格式：
第一行两个整数 n,m n , m ，含义如题
接下来一行 n n 个整数，描述 f1 f 1 到 fn f n
接下来一行 m m 个整数，描述 w1 w 1 到 wn w n

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

---

解法

像这种题，扫一扫右端点，然后线段树维护当前右端点时，左端点选 i i 的贡献是多少就好了
反正相同套路的题很多

---

下面是代码

#include 
#include 
#include 
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() ;
}