CodeForces 620E New Year Tree (DFS序+线段树)

做法:

因为只有60种颜色,所以可以状压颜色,这样的话就可以把区间颜色合并的操作看为“或”,线段树随便维护一下就可以了。
WA了无数发才想起来 ”1LL << 一个东西“ 和 “1 << 一个long long 的东西” 是不一样的(打多校的时候还WA过一发)……贼气.

代码:

#include
using namespace std;
const int maxn=400005;
struct edge{
    int to,nxt;
}e[maxn<<1];
struct node{
    int l,r;
    unsigned long long num,tag;
}tr[maxn<<2];
int L[maxn],R[maxn],rnk[maxn];
int n,m,ecnt;
int head[maxn];
unsigned long long tmp[maxn];
void ins(int u,int v){
    e[ecnt].to=v;
    e[ecnt].nxt=head[u];
    head[u]=ecnt++;
}
int cnt;
void dfs(int u,int fa){
    L[u]=++cnt;
    rnk[cnt]=u;
    for(int i=head[u];~i;i=e[i].nxt){
        if(e[i].to==fa) continue;
        dfs(e[i].to,u);
    }
    R[u]=cnt;
}
void push_up(int k){tr[k].num=tr[k<<1].num|tr[k<<1|1].num;}
void push_down(int k){
    if(tr[k].tag){
        tr[k<<1].num=tr[k<<1|1].num=tr[k].tag;
        tr[k<<1].tag=tr[k<<1|1].tag=tr[k].tag;
        tr[k].tag=0;
    }
}
void build(int k,int l,int r){
    tr[k].l=l,tr[k].r=r;
    if(l==r){
        tr[k].num=1LL<0;
        return;
    }
    int mid=(l+r)>>1;
    build(k<<1,l,mid),build(k<<1|1,mid+1,r);
    push_up(k);
}
void update(int k,int L,int R,unsigned long long x){
    int l=tr[k].l,r=tr[k].r;
    if(l==r){tr[k].num=x;return;}
    if(L<=l&&R>=r){tr[k].tag=tr[k].num=x;return;}
    int mid=(l+r)>>1;
    push_down(k);
    if(R<=mid) update(k<<1,L,R,x);
    else if(L>mid) update(k<<1|1,L,R,x);
    else update(k<<1,L,R,x),update(k<<1|1,L,R,x);
    push_up(k);
}
unsigned long long query(int k,int L,int R){
    int l=tr[k].l,r=tr[k].r;
    if(L<=l&&R>=r) return tr[k].num;
    int mid=(l+r)>>1;
    push_down(k);
    if(R<=mid) return query(k<<1,L,R);
    else if(L>mid) return query(k<<1|1,L,R);
    else return query(k<<1,L,R)|query(k<<1|1,L,R);
    push_up(k);
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) {head[i]=-1;scanf("%I64u",&tmp[i]);}
    int uu,vv;
    for(int i=1;iscanf("%d%d",&uu,&vv);
        ins(uu,vv),ins(vv,uu);
    }
    dfs(1,-1);
    build(1,1,n);
    int op,a;
    unsigned long long b;
    for(int i=1;i<=m;i++){
        scanf("%d",&op);
        if(op==1){
            scanf("%d%I64u",&a,&b);
            update(1,L[a],R[a],(1LL<else{
            scanf("%d",&a);
            printf("%d\n",__builtin_popcountll(query(1,L[a],R[a])));
        }
    }
    return 0;
}

你可能感兴趣的:(数据结构,图论)