Coloring Tree

给你一棵N个节点的树(无向无环连通图)，每个节点有一个颜色。你知道树的根节点（树的根节点是一个特殊的节点，是树节点中唯一没有父节点的)。颜色用1到109范围内的整数表示。现在，你需要回答M个查询，每个查询询问以节点s为根的子树节点中，不同颜色的数目。（s会在输入中给出）。

输入格式

第一行包含3个空格分隔的整数 (N M root)表示树的节点数、查询数和根节点。

接下来N行，每行包含着2个空格分隔的整数(a b)表示一条从节点a到节点b的边。

接下来N行, 第i行表示第i个节点的颜色。

接下来M行，每行包含一个整数s。

输出格式

输出M行，每行输出对应查询的结果。

约束条件

0=< M <= 105
1=< N <= 105
1=< root <= N
1=< 节点颜色 <= 109

样例

输入样例

4 2 1
1 2
2 4
2 3
10
20
20
30
1
2
输出样例

3
2
解释

查询1 以1为根的子树就是整个树，包含颜色10 20 20 30， 所以结果是3(10,20 and 30)。

查询2 以2为节点的子树包含颜色20 20 30， 所以答案是2(20 and 30)。

#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <vector>
#include <set>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 100010;

struct Edge {
    int v, next;
    Edge() { }
    Edge(int _v, int _next) : v(_v), next(_next) { }
}edges[2*maxn];
int head[maxn], edge_num;
int color[maxn], ans[maxn], id[maxn];
set<int> ver[maxn];
int n, m, r;

void init_graph() {
    edge_num = 0;
    memset(head, -1, sizeof(head));
}

void add_edge(int u, int v) {
    edges[edge_num].v = v;
    edges[edge_num].next = head[u];
    head[u] = edge_num++;

    edges[edge_num].v = u;
    edges[edge_num].next = head[v];
    head[v] = edge_num++;
}

void merge(int &u, int &v) {
    if(ver[u].size() < ver[v].size()) swap(u, v);
    for(set<int>::iterator it = ver[v].begin(); it != ver[v].end(); ++it) {
        ver[u].insert(*it);
    }
    return ;
}

void dfs(int u, int f) {
    ver[id[u]].insert(color[u]);
    for(int i = head[u]; i != -1; i = edges[i].next) {
        int v = edges[i].v;
        if(v == f) continue;
        dfs(v, u);
        merge(id[u], id[v]);    
    }
    ans[u] = ver[id[u]].size();
    return ;
}


int main() {
    /* Enter your code here. Read input from STDIN. Print output to STDOUT */   
    int u, v;
    init_graph();
    scanf("%d %d %d", &n, &m, &r);
    for(int i = 1; i < n; ++i) {
        scanf("%d %d", &u, &v);
        add_edge(u, v);
    }
    for(int i = 1; i <= n; ++i) {
        id[i] = i;
        scanf("%d", &color[i]);
    }
    dfs(r, -1);
    for(int i = 0; i < m; ++i) {
        scanf("%d", &u);
        printf("%d\n", ans[u]);
    }
    return 0;
}