【线段树】总集（二）

成段更新。需要用到延迟标记(或者说懒惰标记),简单来说就是每次更新的时候不要更新到底,用延迟标记使得更新延迟到下次需要更新or询问到的时候。

HDU 1698 Just a Hook

线段树功能:update:成段替换 (由于只query一次总区间,所以可以直接输出1结点的信息)

#include <iostream>
#include <math.h>
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#define maxn  100000
using namespace std;

int sum[maxn << 2];
int col[maxn << 2];
void PushDown(int rt, int m)
{
    if(col[rt]){
        col[rt << 1] = col[rt << 1 | 1] = col[rt];
        sum[rt << 1] = (m - (m >> 1)) * col[rt];
        sum[rt << 1 | 1] = (m >> 1) * col[rt];
        col[rt] = 0;
    }
}

void built(int l, int r, int rt)
{
    sum[rt] = r - l + 1;
    col[rt] = 0;
    if(l == r) return ;
    int mid = (l + r) >> 1;
    built(l, mid, rt << 1);
    built(mid + 1, r, rt << 1 | 1);
}

void update(int ll, int rr, int add, int l, int r, int rt)
{
    if(ll <= l && r <= rr)
    {
        col[rt] = add;
        sum[rt] = (r - l + 1) * add;
        return ;
    }
    PushDown(rt, r - l+ 1);
    if( l == r ) return ;
    int mid = (l + r) >> 1;
    if(ll <= mid) update(ll, rr, add, l, mid, rt << 1);
    if(rr > mid) update(ll, rr, add, mid + 1, r, rt << 1 | 1);
    sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];
}

int main()
{
    int t;
    while(scanf("%d", &t) != EOF)
    {
        for(int cas = 1; cas <= t; cas++)
        {
            int n;
            scanf("%d", &n);
            built(1, n, 1);
            int m;
            scanf("%d", &m);
            while(m--)
            {
                int a, b, c;
                scanf("%d%d%d", &a, &b, &c);
                update(a,b,c,1,n,1);
            }
            printf("Case %d: The total value of the hook is %d.\n", cas, sum[1]);
        }
    }
    return 0;
}

POJ 2528 Mayor's posters

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <string.h>
using namespace std;


/*题意:在墙上贴海报,海报可以互相覆盖,问最后可以看见几张海报
思路:这题数据范围很大,直接搞超时+超内存,需要离散化:
离散化简单的来说就是只取我们需要的值来用,
比如说区间[1000,2000],[1990,2012] 我们用不到[-∞,999][1001,1989][1991,1999][2001,2011][2013,+∞]这些值,
所以我只需要1000,1990,2000,2012就够了,将其分别映射到0,1,2,3,在于复杂度就大大的降下来了
所以离散化要保存所有需要用到的值,排序后,分别映射到1~n,这样复杂度就会小很多很多
而这题的难点在于每个数字其实表示的是一个单位长度(并且一个点),
这样普通的离散化会造成许多错误(包括我以前的代码,poj这题数据奇弱)
给出下面两个简单的例子应该能体现普通离散化的缺陷:
1-10 1-4 5-10
1-10 1-4 6-10
为了解决这种缺陷,我们可以在排序后的数组上加些处理,比如说[1,2,6,10]
如果相邻数字间距大于1的话,在其中加上任意一个数字,比如加成[1,2,3,6,7,10],
然后再做线段树就好了.
线段树功能:update:成段替换 query:简单hash
*/

const int maxn = 11111;
int li[maxn], ri[maxn];
int col[maxn << 4];
bool Hash[maxn];
int X[maxn* 3];
int cnt;

void pushdown(int rt){
    if(col[rt] != -1){
        col[rt<<1] = col[rt<<1|1] = col[rt];
        col[rt] = -1;
    }
}

//void build(int l, int r, int rt){
//    add[rt] = 0;
//    if(l == r){
//        scanf("%lld", &sum[rt]);
//        return ;
//    }
//    int m = (l + r) >> 1;
//    build(l, m, rt<<1);
//    build(m+1, r, rt<<1|1);
//    sum[rt] = sum[rt<<1] + sum[rt<<1|1];
//}

void update(int L, int R, int color, int l, int r, int rt){
    if(L <= l && r <= R){
        col[rt] = color;
        return ;
    }
    pushdown(rt);
    int m = (l + r) >> 1;
    if(L <= m) update(L, R, color, l, m, rt<<1);
    if(m < R) update(L, R, color, m+1, r, rt<<1|1);
}

void query(int l, int r, int rt){
    if(col[rt] != -1){
        if(!Hash[col[rt]]) cnt++;
        Hash[col[rt]] = true;
        return ;
    }
    if(l == r) return ;
    int m = (l + r) >> 1;
    query(l, m, rt<<1);
    query(m+1, r, rt<<1|1);
}

int Bin(int key, int n, int X[]){  //二分法找到离散，排序后原key对应的序号
    int l = 0, r = n - 1;
    while(l <= r){
        int m = (l + r) >> 1;
        if(X[m] == key) return m;
        if(X[m] < key) l = m + 1;
        else r = m - 1;
    }
    return -1;
}

int main(){
    int T;
    scanf("%d", &T);
    while(T--){
        int n, nn = 0;
        scanf("%d", &n);
        for(int i = 0; i < n; i++){ //input, all numbers put into X array;
            scanf("%d%d", &li[i], &ri[i]);
            X[nn++] = li[i];
            X[nn++] = ri[i];
        }

        sort(X, X + nn);

        int m = 1;
        for(int i = 1; i < nn; i++) //delete repeating numbers
            if(X[i] != X[i-1]) X[m++] = X[i];

        for(int i = m -1; i > 0; i--)   //如果相邻数字间距大于1的话,在其中加上任意一个数字
            if(X[i] != X[i-1] + 1) X[m++] = X[i] + 1;

        sort(X, X + m);
        memset(col, -1, sizeof(col));

        for(int i = 0; i < n; i++){ //color
            int l = Bin(li[i], m, X);
            int r = Bin(ri[i], m, X);
            update(l, r, i, 0, m, 1);   //m-1?
        }

        cnt = 0;
        memset(Hash, false, sizeof(Hash));
        query(0, m, 1);   //m-1?
        printf("%d\n", cnt);
    }
    return 0;
}