scu || problem source 組成 ref oid clu mem

2243: [SDOI2011]染色

Time Limit: 20 Sec Memory Limit: 512 MB
Submit: 8800 Solved: 3305
[Submit][Status][Discuss]

Description

給定一棵有n個節點的無根樹和m個操作，操作有2類： 1、將節點a到節點b路徑上所有點都染成顏色c； 2、詢問節點a到節點b路徑上的顏色段數量（連續相同顏色被認為是同一段）， 如“112221”由3段組成：“11”、“222”和“1”。 請你寫一個程序依次完成這m個操作。

Input

第一行包含2個整數n和m，分別表示節點數和操作數； 第二行包含n個正整數表示n個節點的初始顏色 下面 行每行包含兩個整數x和y，表示x和y之間有一條無向邊。 下面 行每行描述一個操作： “C a b c”表示這是一個染色操作，把節點a到節點b路徑上所有點（包括a和b）都染成顏色c； “Q a b”表示這是一個詢問操作，詢問節點a到節點b（包括a和b）路徑上的顏色段數量。

Output

對於每個詢問操作，輸出一行答案。

Sample Input

6 5
2 2 1 2 1 1
1 2
1 3

2 4
2 5
2 6
Q 3 5
C 2 1 1
Q 3 5
C 5 1 2
Q 3 5

Sample Output

3
1
2

HINT

數N<=10^5，操作數M<=10^5，所有的顏色C為整數且在[0, 10^9]之間。

Source

第一輪day1

代碼：

//線段樹區間染色查詢區間顏色段數量，樹剖之後還要考慮兩段銜接之後顏色段是否減少。
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace
 
 std;
const int MAXN=100009;
int n,m,fa[MAXN],son[MAXN],lev[MAXN],top[MAXN],id[MAXN],head[MAXN],size[MAXN],val[MAXN];
int tot,cnt,col[MAXN<<2],tag[MAXN<<2],lef[MAXN<<2],rig[MAXN<<2],a[MAXN],nowr,nowl;
struct Edge { int to,next; }edge[MAXN<<1];
void init()
{
    tot=cnt=0;
    memset(head,-1,sizeof(head));
    for(int i=1;i<=n;i++) fa[i]=top[i]=i;
}
void addedge(int x,int y)
{
    edge[tot].to=y;edge[tot].next=head[x];
    head[x]=tot++;
    edge[tot].to=x;edge[tot].next=head[y];
    head[y]=tot++;
}
void dfs1(int x,int d)
{
    size[x]=1;
    lev[x]=d;
    son[x]=0;
    for(int i=head[x];i!=-1;i=edge[i].next){
        int y=edge[i].to;
        if(y==fa[x]) continue;
        fa[y]=x;
        dfs1(y,d+1);
        size[x]+=size[y];
        if(size[son[x]]<size[y]) son[x]=y;
    }
}
void dfs2(int x,int tp)
{
    id[x]=++cnt;
    top[x]=tp;
    if(son[x]) dfs2(son[x],tp);
    for(int i=head[x];i!=-1;i=edge[i].next){
        int y=edge[i].to;
        if(y==fa[x]||y==son[x]) continue;
        dfs2(y,y);
    }
}
void pushup(int rt)
{
    lef[rt]=lef[rt<<1];
    rig[rt]=rig[rt<<1|1];
    col[rt]=col[rt<<1]+col[rt<<1|1];
    if(rig[rt<<1]==lef[rt<<1|1]) col[rt]--;
}
void build(int rt,int l,int r)
{
    tag[rt]=-1;
    if(l==r){
        col[rt]=1;
        lef[rt]=rig[rt]=val[l];
        return;
    }
    int mid=(l+r)>>1;
    build(rt<<1,l,mid);
    build(rt<<1|1,mid+1,r);
    pushup(rt);
}
void pushdown(int rt)
{
    if(tag[rt]!=-1){
        tag[rt<<1]=tag[rt<<1|1]=tag[rt];
        lef[rt<<1]=lef[rt<<1|1]=rig[rt<<1]=rig[rt<<1|1]=tag[rt];
        col[rt<<1]=col[rt<<1|1]=1;
        tag[rt]=-1;
    }
}
void update(int ql,int qr,int c,int rt,int l,int r)
{
    if(ql<=l&&qr>=r){
        tag[rt]=lef[rt]=rig[rt]=c;
        col[rt]=1;
        return;
    }
    pushdown(rt);
    int mid=(l+r)>>1;
    if(ql<=mid) update(ql,qr,c,rt<<1,l,mid);
    if(qr>mid) update(ql,qr,c,rt<<1|1,mid+1,r);
    pushup(rt);
}
void Update(int ql,int qr,int c)
{
    int ltp=top[ql],rtp=top[qr];
    while(ltp!=rtp){
        if(lev[rtp]<lev[ltp]){
            swap(ql,qr);
            swap(ltp,rtp);
        }
        update(id[rtp],id[qr],c,1,1,cnt);
        qr=fa[rtp];
        rtp=top[qr];
    }
    if(lev[qr]<lev[ql]) swap(qr,ql);
    update(id[ql],id[qr],c,1,1,cnt);
}
int query(int ql,int qr,int rt,int l,int r)
{
    if(ql==l) nowl=lef[rt];
    if(qr==r) nowr=rig[rt];
    if(ql<=l&&qr>=r) return col[rt];
    pushdown(rt);
    int mid=(l+r)>>1,s=0,lc=-1,rc=-1;
    if(ql<=mid){
        s+=query(ql,qr,rt<<1,l,mid);
        lc=rig[rt<<1];
    }
    if(qr>mid){
        s+=query(ql,qr,rt<<1|1,mid+1,r);
        rc=lef[rt<<1|1];
    }
    if(lc==rc&&lc!=-1) s--;
    return s;
}
int Query(int ql,int qr)
{
    int s=0,ltp=top[ql],rtp=top[qr],lastl=-1,lastr=-1;
    nowl=-1,nowr=-1;
    while(ltp!=rtp){
        if(lev[rtp]<lev[ltp]){
            swap(ql,qr);
            swap(ltp,rtp);
            swap(lastl,lastr);
        }
        s+=query(id[rtp],id[qr],1,1,cnt);
        if(nowr==lastr&&lastr!=-1) s--;
        lastr=nowl;
        qr=fa[rtp];
        rtp=top[qr];
    }
    if(lev[qr]<lev[ql]) { swap(ql,qr);swap(lastl,lastr); }
    s+=query(id[ql],id[qr],1,1,cnt);
    if(nowl==lastl&&lastl!=-1) s--;
    if(nowr==lastr&&lastr!=-1) s--;
    return s;
}
int main()
{
    //freopen("in.txt","r",stdin);
    scanf("%d%d",&n,&m);
    init();
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<n;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        addedge(x,y);
    }
    dfs1(1,1);
    dfs2(1,1);
    for(int i=1;i<=n;i++) val[id[i]]=a[i];
    build(1,1,cnt);
    while(m--){
        char ch[3];
        int x,y,z;
        scanf("%s",ch);
        if(ch[0]==‘Q‘){
            scanf("%d%d",&x,&y);
            printf("%d\n",Query(x,y));
        }else{
            scanf("%d%d%d",&x,&y,&z);
            Update(x,y,z);
        }
    }
    return 0;
}

bzoj 2243