zstu3999——邻接表

Description

多组测试数据  给你一棵树，n个点(n<=10000),树的根为root(root<=n),然后有Q个询问，每个询问输入一个数x，求以x为根的子树有多少个点？

Input

每组的第一行是一个整数n和一个整数root  接下来n-1行是这棵树的n-1条边，每行两个数a b，代表a点和b点之间有一条边  然后是一个整数Q，代表询问的数量  最后Q行每行是一个整数(Q<=n)

Output

对于每个询问，输出以当前点为根的子树的节点总数

Sample Input

3 1
1 2
1 3
1
1

Sample Output

3

HINT

 

Source

No_stop

Submit Status Web Board

 大意：用vector来保存每一个节点的儿子节点，dfs大法好

学习了几个vector的处理函数

vector.begin()开始的点 ，vector.end()最后的点，vector.clear()清空，这个很重要~~

还有迭代器相当于一个指针 vector :: iterator i定义i为vector的一个指针

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn = 10100;
int mark[maxn],cot[maxn];
vector<int> G[10100];
int dfs(int root)
{
    mark[root] = 1;
    if(G[root].size() == 0){
        cot[root] = 1;
        return 1;
    }
    vector<int>:: iterator endd = G[root].end();
    for(vector<int>:: iterator i = G[root].begin(); i!= endd; i++){
        if(mark[*i]) continue;
        cot[root] += dfs(*i);
    }
    cot[root]++;
    return cot[root];
}
int main()
{
    int n,m,q,x;
    int a,b;
    while(~scanf("%d%d",&n,&m)){
        memset(mark,0,sizeof(mark));
        memset(cot,0,sizeof(cot));
        for(int i = 1; i < n ;i++){
            scanf("%d%d",&a,&b);
            G[a].push_back(b);
            G[b].push_back(a);
        }
        dfs(m);
      scanf("%d",&q);
      for(int i = 1; i <= q; i++){
          scanf("%d",&x);
          printf("%d\n",cot[x]);
      }
      for(int i = 1; i <= n;i++)
          G[i].clear();
    }
    return 0;
}