1. 程式人生 > >線段樹染色問題(區間一共有多少種不同顏色)poj2777

線段樹染色問題(區間一共有多少種不同顏色)poj2777

 

描述

選擇問題解決和程式設計作為選修課程,您需要解決各種問題。在這裡,我們遇到了一個新問題。 

有一個很長的板長L釐米,L是正整數,所以我們可以將板均分為L段,從左到右標記為1,2,...... L,每個為1釐米長。現在我們必須為電路板上色 - 只有一種顏色的一個部分。我們可以在電路板上執行以下兩項操作:

1。“CAB C”將電路板從A段染色到B段,顏色為C. 
2.“PA B”輸出A段和B段之間繪製的不同顏色的數量(包括)。 

在我們的日常生活中,我們用很少的詞來描述顏色(紅色,綠色,藍色,黃色......),因此您可以假設不同顏色T的總數非常小。為了簡單起見,我們將顏色的名稱表示為顏色1,顏色2,...顏色T.開始時,板上塗有顏色1.現在剩下的問題留給你了。 

輸入

第一行輸入包含L(1 <= L <= 100000),T(1 <= T <= 30)和O(1 <= O <= 100000)。這裡O表示操作次數。在O行之後,每個包含“CAB C”或“PA B”(這裡A,B,C是整數,A可以大於B)作為先前定義的操作。

產量

輸出操作的輸出結果按順序,每行包含一個數字。

樣本輸入

<span style="color:#000000">2 2 4
C 1 1 2
P 1 2
C 2 2 2
P 1 2
</span>

樣本輸出

<span style="color:#000000">2
1</span>

 首先本題的區間是一個一個的整數,所以不用乘2考慮開區間閉區間的問題了.所以這裡的線段樹維護一個color資訊,其中color>0表示節點所指的區間都有某種顏色color, color=-1 表示該節點的子節點的顏色不一致.

       build操作: 遞迴建樹,並且所有color初始設定為1.

       PushUp操作: 根據子節點的color更新i節點的color.如果子節點color一致,那麼i節點color與子節點color一樣.否則i節點color為-1.

       PushDown操作: color>0時才把color傳遞到下面去.PushDown操作可以不用改變i(父)節點的color.

       update操作: 如果[ql, qr]包括了[l, r] ,那麼就直接置新color值,否則先PushDonw,然後分段更新color,再PushUp.

       query操作: 本操作主要是用一個cnt全域性變數反饋[ql, qr]區間內一共有多少個不同的color.每次查詢前都初始化vis[30]為false,出現一種顏色就vis[i]=true.如果[ql, qr]包括了[l, r]區間,那麼直接更新vis,否則PushDown後,分段查詢.
 

<span style="font-size:18px;">//一次AC
#include <iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
#define lson i*2,l,m
#define rson i*2+1,m+1,r
const int MAXN = 100000 + 100;
bool vis[35];
int cnt;//用來計數顏色
struct IntervalTree
{
    int color[MAXN * 4];
 
    void build(int i, int l, int r)
    {
        color[i] = 1;
        if(l == r) return ;
        int m = (l + r) / 2;
        build(lson);
        build(rson);
    }
 
    void PushDown(int i)
    {
        if(color[i] > 0)
            color[i * 2] = color[i * 2 + 1] = color[i];
    }
 
    void PushUp(int i)
    {
        if(color[i * 2] == -1 || color[i * 2 + 1] == -1)
            color[i] = -1;
        else if(color[i * 2] == color[i * 2 + 1])
            color[i] = color[i * 2];
        else
            color[i] = -1;
    }
 
    void update(int ql, int qr, int v, int i, int l, int r)
    {
        if(ql <= l && r <= qr)
        {
            color[i] = v;
            return ;
        }
        PushDown(i);
        int m = (l + r) / 2;
        if(ql <= m) update(ql, qr, v, lson);
        if(m < qr) update(ql, qr, v, rson);
        PushUp(i);
    }
 
    void query(int ql, int qr, int i, int l, int r)
    {
        if(color[i] > 0)
        {
            if(vis[color[i]] == false)
                cnt++;
            vis[color[i]] = true;
            return ;
        }
        //PushDown(i);
        int m = (l + r) / 2;
        if(ql <= m) query(ql, qr, lson);
        if(m < qr) query(ql, qr, rson);
    }
};
IntervalTree T;
int main()
{
    int n, t, q;
    while(scanf("%d%d%d", &n, &t, &q) == 3)
    {
        T.build(1, 1, n);
        while(q--)
        {
            char str[10];
            scanf("%s", str);
            if(str[0] == 'C')
            {
                int x, y, z;
                scanf("%d%d%d", &x, &y, &z);
                T.update(x, y, z, 1, 1, n);
            }
            else if(str[0] == 'P')
            {
                int x, y;
                scanf("%d%d", &x, &y);
                memset(vis, 0, sizeof(vis));
                cnt = 0;
                T.query(x, y, 1, 1, n);
                printf("%d\n", cnt);
            }
        }
    }
    return 0;
}
</span>