ZOJ-1610 Count the Colors ( 线段树 )

题目链接: http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=1610

Description

Painting some colored segments on a line, some previously painted segments may be covered by some the subsequent ones.

Your task is counting the segments of different colors you can see at last.

Input

The first line of each data set contains exactly one integer n, 1 <= n <= 8000, equal to the number of colored segments.

Each of the following n lines consists of exactly 3 nonnegative integers separated by single spaces:

x1 x2 c

x1 and x2 indicate the left endpoint and right endpoint of the segment, c indicates the color of the segment.

All the numbers are in the range [0, 8000], and they are all integers.

Input may contain several data set, process to the end of file.

Output

Each line of the output should contain a color index that can be seen from the top, following the count of the segments of this color, they should be printed according to the color index.

If some color can't be seen, you shouldn't print it.

Print a blank line after every dataset.

Sample Input

5
0 4 4
0 3 1
3 4 2
0 2 2
0 2 3
4
0 1 1
3 4 1
1 3 2
1 3 1
6
0 1 0
1 2 1
2 3 1
1 2 0
2 3 0
1 2 1

Sample Output

1 1
2 1
3 1

1 1

0 2
1 1

在[0, 8000]涂色，后涂的会覆盖前面涂的，求每种颜色最后能分辨的的区块有多少个。

建树时，树的单位是区间，而不是点 如[0, 8000]的左右儿子应该分别是[0, 4000], [4000,8000];

询问时，需要判断相邻区间的颜色是否相同 用temp记录前一区间所记录的颜色来解决这个问题。

#include<iostream>
#include<cstring>
using namespace std;

int color[8010];
int mark[20000];
int temp;

void Build( int i, int l, int r ){
    mark[i] = -1;//-1表示没有涂

    if( l + 1 == r ) return;

    int mid = ( l + r ) >> 1;
    Build( i << 1, l , mid );
    Build( ( i << 1 ) | 1, mid, r );
}

void Insert( int i, int l, int r, int z, int y, int c ){
    if( l == r ) return;
    if( z <= l && y >= r ){
        mark[i] = c;
        return;
    }
    if( mark[i] == c ) return;
    if( mark[i] >= 0 ){
        mark[i << 1] = mark[i];
        mark[( i << 1) | 1] = mark[i];
        mark[i] = -2;//该区间含有多个颜色
    }
    int mid = ( l + r ) >> 1;
    if( y <= mid ) Insert( i << 1, l, mid, z, y, c );
    else if( z >= mid ) Insert( ( i << 1 ) | 1, mid, r, z, y, c );
    else{
        Insert( i << 1, l, mid, z, mid, c );
        Insert( ( i << 1 ) | 1, mid, r, mid, y, c );
    }
    mark[i] = -2;
}

void Queue( int i, int l, int r){
    if( mark[i] == -1 ){
        temp = -1;
        return;
    }
    if( mark[i] >= 0 ){
        if( temp != mark[i] ){
            temp = mark[i];//记录前一段的颜色
            color[mark[i]]++;
        }
        return;
    }
    if( l + 1 != r ){
        int mid = ( l + r ) >> 1;
        Queue( i << 1, l, mid );
        Queue( ( i << 1) | 1, mid, r ); 
    }
}

int main(){
    ios::sync_with_stdio( false );

    int n;
    while( cin >> n ){

        Build( 1, 0, 8000 );
        memset( color, 0, sizeof( color ) );

        int a, b, c, max = 0;
        for( int i = 1; i <= n; i++ ){
            cin >> a >> b >> c;
            Insert( 1, 0, 8000, a, b, c );
            max = c > max ? c : max;
        }

        temp = -1;
        Queue( 1, 0, 8000 );

        for( int i = 0; i <= max ; i++ )
            if( color[i] ) cout << i << " " << color[i] << endl;

        cout << endl;
    }

    return 0;
}