HDU4107 Gangster 线段树 段更新

转载注明出处 http://blog.csdn.net/moedane

 

 

传送门 http://acm.hdu.edu.cn/showproblem.php?pid=4107

题意

代表有n个数（区间为[1,n]），m种操作。每个操作是给区间[l,r]加上一个数c，如果这个区间里面某个数大于等于给出的界限P，则加上2*c。输出全部操作结束之后最终的数列。

思路

线段树断更新。要记录的信息有区间最大值Max、区间最小值Min、区间累加的值addv。

要判断是否大于等于界限，则只要判断Min是否大于等于P、或者Max是否小于P。

if(Min >= P) add(2*c);

if(Max < p) add(c);

其他情况只需要往下再跑一层，直到达到上面的条件则可以做出相应操作。

这道题有点恶心的地方就是时间卡得紧，我把大部分预算换成位运算加上用输入挂才能950+ms勉强过。

代码

#include <iostream>
#include <cstring>
#include <cstdio>
#include <string>
#include <algorithm>
#include <cmath>
#include <ctime>
#include <vector>
#include <queue>
#include <stack>
#include <fstream>
#include <map>
#include <set>
#define bug puts("here");

using namespace std;

typedef long long ll;

const int maxn = 2 * 101000;
const ll mod = 1e9+7;
const int inf = 0x3f3f3f3f;
const double PI = atan(1.0) * 4.0;
const double eps = 1e-8;

int n,m,P;

struct Node
{
    int o,l,r;
    int Max,Min;
    int addv;
    int lenth() {return r - l + 1;}
    int mid() {return l + (r-l) / 2;}
    bool in(int L,int R) {return L <= l && R >= r;}
};

struct T
{
    Node t[maxn * 4];
    void build(int o,int l,int r)
    {
        t[o].l = l;
        t[o].r = r;
        t[o].o = o;
        t[o].addv = t[o].Min = t[o].Max = 0;
        if(l == r) return;
        build(o<<1, l , t[o].mid());
        build(o<<1|1, t[o].mid() + 1 , r);
    }

    void pushdown(int o)
    {
        if(t[o].addv && t[o].lenth() > 1)
        {
            t[o<<1].addv += t[o].addv;
            t[o<<1|1].addv += t[o].addv;
            t[o<<1|1].Min += t[o].addv;
            t[o<<1].Min += t[o].addv;
            t[o<<1|1].Max += t[o].addv;
            t[o<<1].Max += t[o].addv;
            t[o].addv = 0;
        }
    }

    void add(int o,int l,int r,int v)
    {
        if(t[o].in(l,r)){
            if(t[o].Max < P){
                t[o].addv += v;
                t[o].Max += v;
                t[o].Min += v;
                return;
            }
            if(t[o].Min >= P) {
                t[o].addv += v << 1;
                t[o].Max += v << 1;
                t[o].Min += v << 1;
                return;
            }
        }
        pushdown(o);
        if(l <= t[o].mid()) add(o<<1, l, r, v);
        if(r > t[o].mid()) add(o<<1|1, l , r, v);
        t[o].Min = min(t[o<<1].Min , t[o<<1|1].Min);
        t[o].Max = max(t[o<<1].Max , t[o<<1|1].Max);
        return;
    }

    int query(int o,int x)
    {
        if(t[o].lenth() == 1 && t[o].l == x){
            return t[o].addv;
        }
        pushdown(o);
        if(x <= t[o].mid()) query(o<<1, x);
        else query(o<<1|1, x);
    }
}tree;

inline void scan(int &n)
{
	char cc;
	for (; cc = getchar(), cc<'0' || cc>'9';);
	n = cc - '0';
	for (; cc = getchar(), cc >= '0'&&cc <= '9';)
		n = n * 10 + cc - '0';
}

int main()
{
    while(~scanf("%d%d%d",&n,&m,&P))
    {
        tree.build(1,1,n);
        for(int i=0;i<m;i++)
        {
            int l,r,c;
            scan(l);
            scan(r);
            scan(c);
            tree.add(1,l,r,c);

        }
        for(int i=1;i<n;i++){
            printf("%d ",tree.query(1,i));
        }
        printf("%d\n",tree.query(1,n));
    }
}