線段樹染色問題(區間一共有多少種不同顏色)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>