FZU 2079 最大获利（线段树+DP）

最新推荐文章于 2020-04-02 16:53:23 发布

weixin_33862041

最新推荐文章于 2020-04-02 16:53:23 发布

阅读量81

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/hlmark/p/4125817.html

Description

Sean准备投资一些项目。有n个投资项目，投资第i个项目需要花费Ci元。Sean发现如果投资了某些编号连续的项目就能赚得一定的钱。现在给出m组连续的项目和每组能赚得的钱，请问采取最优的投资策略的最大获利是多少？

样例最佳策略是全部项目都投资，然后第1，2组都满足了，获利为2+2-3=1。最佳策略可能是不投资，即最大获利为0。

Input

每组数据第一行输入两个整数N和M , N表示项目数，M表示能获利的连续的项目组数，1 <= N <= 20000,1 <= M <= 20000 , 接下来一行输入N个数，表示每个项目投资需要的花费Ci，1<=Ci<=10^9。

接下来m行，每行3个数Li,Ri,Pi，1<=Li<=Ri<=N,1<=Pi<=10^9,表示投资从第Li到第Ri这些连续的项目的获利。每组连续投资的获利互不影响，如果投资的多组连续投资都包含项目i，项目i只需要投资一次。

Output

对于每组数据输出一行，即最大获利。

Sample Input

3 2

1 1 1

1 2 2

2 3 2

Sample Output

设dp[i]表示，前i个项目的最大获利。

转移方程： dp[i] = max( dp[i-1] , dp[j-1] + w[j][i] ) , ( 1<=j <=n ).

枚举状态费用O（n） , 用线段树优化取最值，更新费用为O(log n ) , 总复杂度为O（n log n ）。

线段树的每个节点i表示，从i连续投资到当前节点的最大利润

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <cmath>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <stack>
#include <algorithm>
using namespace std;

typedef long long LL;
const int N = 200010;
const int inf = 1e9+7;
const double PI = acos(-1.0);
const double eps = 1e-6 ;

#define root 1,n,1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define lr rt<<1
#define rr rt<<1|1

int n , m ;
struct node {
    int l , r , w ;
    bool operator < ( const node &a ) const {
        if( r != a.r ) return r < a.r ;
        else return l > a.l ;
    }
}e[N];

LL dp[N] , cost[N] ;

void init(){
    memset( dp , 0 , sizeof dp );
}

LL date[N<<2] ,lazy[N<<2];

void build( int l , int r , int rt ) {
    date[rt] = lazy[rt] = 0 ;
    if( l == r ) return ;
    int mid = ( l + r ) >> 1 ;
    build(lson); build(rson);
}

void Down( int l , int r , int rt ) {
    if( l == r ) return  ;
    if( lazy[rt] != 0 ) {
        date[lr]+= lazy[rt] , lazy[lr] += lazy[rt];
        date[rr]+= lazy[rt] , lazy[rr] += lazy[rt];
        lazy[rt] = 0 ;
    }
}

void Up( int rt ) { date[rt] = max( date[lr] , date[rr] ); }

void update( int l , int r , int rt , int L , int R , LL val) {

    if( L == l && r == R ) {
        date[rt] += val , lazy[rt] += val ; return ;
    }
    Down(l,r,rt);
    int mid = (l+r) >> 1;
    if( R <= mid ) update(lson,L,R,val);
    else if( L > mid ) update(rson,L,R,val);
    else update(lson,L,mid,val) , update(rson,mid+1,R,val);
    Up(rt);
}

LL query( int l , int r , int rt , int L , int R ) {

    if( L == l && r == R ) {
        return date[rt];
    }
    Down(l,r,rt);
    int mid = (l+r)>>1;
    if( R <= mid ) return query(lson,L,R);
    else if( L > mid ) return query(rson,L,R);
    else return max( query(lson,L,mid),query(rson,mid+1,R) );
}

void run () {
    init() ;
    build(root);
    for( int i = 1 ; i <= n ; ++i ){
        scanf("%I64d",&cost[i]) ;
    }
    for( int i = 0 ; i < m ; ++i ) {
        scanf("%d%d%d",&e[i].l,&e[i].r,&e[i].w);
    }
    sort( e , e + m );
    int head = 0 ;
    for( int i = 1 ; i <= n ; ++i ) {
        update( root , i  , i  , dp[i-1] );
        update( root , 1 , i , -cost[i] );
        while( head < m && e[head].r <= i )
            update( root , 1 , e[head].l , e[head].w ) , head++ ;
        dp[i] = max( dp[i-1] , query( root , 1 , i ) );
    }
    printf("%I64d\n",dp[n]);
}
int main()
{
    #ifdef LOCAL
        freopen("in.txt","r",stdin);
    #endif // LOCAL
    ios::sync_with_stdio(false);
    while( ~scanf("%d%d",&n,&m) )run() ;
}