Educational Codeforces Round 6 E dfs序+线段树

Posted 2020-08-17 天翎月

题意：给出一颗有根树的构造和一开始每个点的颜色 有两种操作 1 ： 给定点的子树群体涂色 2 ： 求给定点的子树中有多少种颜色

比较容易想到dfs序+线段树去做

dfs序是很久以前看的bilibili上电子科技大学发的视频学习的 将一颗树通过dfs编号的方式 使每个点的子树的编号连在一起作为相连的区间 就可以配合线段树搞子树

因为以前好像听说过 线段树可以解决一种区间修改和查询区间中不同的xx个数...所以一下子就想到了...

但是我不会写线段树..只会最简单的单点修改区间查询...不会用延迟标记...所以拿出书现学了一发..

问学长怎么记录不同的颜色个数 学长就机智的告诉了我用longlong来搞..

虽然知道了思路..还是写了一天多..

dfs序 做出每个点的编号 并且记录每个点的子树的编号的左右区间 

初始进行crea的时候 进行赋值并且pushup

利用longlong的64位来存有哪种颜色 利用或来做转移

1L<<50 什么的 好像左移过了多少就崩了..要在外边定义L的变量..哭..

写线段树好累...(脸上挂满泪痕)...不过桃桃的小粉红敲起来好舒服...

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<map>
#include<math.h>
#include<iostream>
#include<queue>
#include<string>
using namespace std;
#define L long long
int n , m ;
struct node{
    int l,r;
    L ma;
};
int a[400050];
node tree[400050*8];
vector<int >q[400050];
int id[400050];
L mark[400050*8];
int cnt ;
struct no{
    int l,r;
};
no zg[400050];
int fx[400050];
void dfs(int u){
    id[u] = ++cnt;
    fx[cnt] = u;
    zg[u].l = cnt;
    for(int i=0;i<q[u].size();i++){
        int v = q[u][i];
        if(id[v] == -1){
           dfs(v);
        }
    }
    zg[u].r = cnt;
}
void pushup(int root){
    tree[root].ma = tree[root*2].ma | tree[root*2+1].ma;
}
void pushdown(int root){
    if(mark[root] != -1){
        mark[root*2] = mark[root*2+1] = mark[root];
        tree[root].ma = mark[root];
        tree[root*2].ma = tree[root*2+1].ma = mark[root];
        mark[root] = -1;
    }
}
void crea(int root ,int l,int r){
    tree[root].l = l;
    tree[root].r = r;
    if(l == r){
        L res = 1;
        res <<= a[fx[l]];
        tree[root].ma = (res);
        return ;
    }
    int m = (l + r) >> 1;
    crea(root*2,l,m);
    crea(root*2+1,m+1,r);
    pushup(root);
}
void upda(int root , int l , int r ,int c){
    if(tree[root].r < l || tree[root].l > r){
        return ;
    }
    if(tree[root].r <= r && tree[root].l >= l){
        L res = 1;
        res <<= c;
        mark[root] = (res);
        pushdown(root);
        return ;
    }
    pushdown(root);
    upda(root*2,l,r,c);
    upda(root*2+1,l,r,c);
    pushup(root);
}
L query(int root ,int l , int r){
    pushdown(root);
    if(tree[root].r < l || tree[root].l > r){
        return 0;
    }
    if(tree[root].r <= r && tree[root].l >= l){
        return tree[root].ma;
    }
    L ans = 0;
    ans |= query(root*2,l,r);
    ans |= query(root*2+1,l,r);
    pushup(root);
    return ans ;
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        q[i].clear();
    }
    cnt = 0;
    memset(id,-1,sizeof(id));
    for(int i=1;i<=n-1;i++){
        int u,v;
        scanf("%d%d",&u,&v);
        q[v].push_back(u);
        q[u].push_back(v);
    }
    memset(mark,-1,sizeof(mark));
    dfs(1);
    crea(1,1,cnt);
    for(int i=1;i<=m;i++){
        int k ;
        scanf("%d",&k);
        if(k == 1){
            int v,c;
            scanf("%d%d",&v,&c);
            int ll = zg[v].l;
            int rr = zg[v].r;
            upda(1,ll,rr,c);
        }
        else {
            int v;
            scanf("%d",&v);
            int ll = zg[v].l;
            int rr = zg[v].r;
            L res = query(1,ll,rr);
            int ans = 0;
            while(res > 0){
                ans += (res %2);
                res >>= 1;
            }
            printf("%d\n",ans);
        }
    }
}