【線段樹】Count the Colors

来源:互联网 发布:ubuntu xfce4 美化 编辑:程序博客网 时间:2024/05/16 08:23

Painting some colored segments on a line, some previously paintedsegments may be covered by some the subsequent ones.Your task is counting the segments of different colors you cansee at last.InputThe 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 nonnegativeintegers separated by single spaces:x1 x2 cx1 and x2 indicate the left endpoint and right endpoint of thesegment, c indicates the color of the segment.All the numbers are in the range [0, 8000], and they are allintegers.Input may contain several data set, process to the end of file.OutputEach line of the output should contain a color index that canbe seen from the top, following the count of the segments ofthis 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 Input50 4 40 3 13 4 20 2 20 2 340 1 13 4 11 3 21 3 160 1 01 2 12 3 11 2 02 3 01 2 1Sample Output1 12 13 11 10 21 1
一道線段樹的入門題。

先維護出所有點的顏色(方法見Mayer's Posters),再進行統計。
詳細細節見程序注釋。

Accode:

#include <cstdio>#include <cstring>#include <cstdlib>#include <bitset>const char fi[] = "zoj1610.in";const char fo[] = "zoj1610.out";const int maxN = 8010;const int MAX = 0x3fffff00;const int MIN = -MAX;struct SegTree {int L, R, lc, rc, col; };SegTree tree[maxN << 2];int col[maxN];int cnt[maxN];int L[maxN], R[maxN];int n, m, tot, N;  void init_file()  {    freopen(fi, "r", stdin);    freopen(fo, "w", stdout);  }    void Build(int L, int R)  {    int Now = ++tot;    tree[Now].L = L;    tree[Now].R = R;    tree[Now].col = -1;    int Mid = (L + R) >> 1;    if (L + 1 < R)    {      tree[Now].lc = tot + 1;      Build(L, Mid);      tree[Now].rc = tot + 1;      Build(Mid, R);    }  }    void insert(int Now, int i)  {    if (tree[Now].col == col[i]) return;    if (L[i] <= tree[Now].L && R[i] >= tree[Now].R)      {tree[Now].col = col[i]; return; }    if (tree[Now].col > -2)    {      tree[tree[Now].lc].col =        tree[tree[Now].rc].col =        tree[Now].col;      tree[Now].col = -2;    }    int Mid = (tree[Now].L + tree[Now].R) >> 1;    if (L[i] < Mid) insert(tree[Now].lc, i);    if (Mid < R[i]) insert(tree[Now].rc, i);    if (tree[tree[Now].lc].col > -2      && tree[tree[Now].rc].col > -2      && tree[tree[Now].lc].col      == tree[tree[Now].rc].col)      tree[Now].col = tree[tree[Now].lc].col;//注意最後經染色可能把整個線段染成同種顏色,//所以這裡維護一下標記。  }    void count(int Now)  {    if (tree[Now].col == -1) return;    if (tree[Now].col > -1)    {      col[tree[Now].L + 1] = tree[Now].col;      if (col[tree[Now].R + 1] < -1)        col[tree[Now].R + 1] = -1;    }//用一個近似離散化的操作,//對每條線段的起始和結尾作上標記,//LR都加一是為了防止0下標,方便統計。    if (tree[Now].col < -1)    {      count(tree[Now].lc);      count(tree[Now].rc);    } //為多色則二分統計。  }    void work()  {    while (scanf("%d", &n) != EOF)    {      tot = 0;      Build(0, maxN - 1);      N = 0;      for (int i = 1; i < n + 1; ++i)      {        scanf("%d%d%d", L + i, R + i, col + i);        N = std::max(col[i], N);        insert(1, i);      }      memset(col, 0xfe, sizeof(col));//先把所有點的顏色賦為一個不可能的值。      memset(cnt, 0, sizeof(cnt));      count(1);      int Last = -1;      for (int i = 1; i < maxN; ++i)       if (col[i] > -2)        {          if (col[i] > -1 && col[i] != Last)            ++cnt[col[i]];//有色且與上一段不重複則統計。          Last = col[i];        }      for (int i = 0; i < N + 1; ++i)        if (cnt[i]) printf("%d %d\n", i, cnt[i]);      printf("\n");    }  }  int main(){  init_file();  work();  exit(0);}