SDNU 1057.树的查询 及其拓展

SDNU 1057.树的查询

Time Limit: 1000 MS Memory Limit: 32768 KB
Total Submission(s): 350 Accepted Submission(s): 61

Description
给定 n(1 <= n <= 1000000), m(1 <= m <= 10) 分别表示一棵树中节点的个数及查询的数量，每个节点的编号为给定的顺序，之后给定每个节点的父节点编号，及 m 个查询，每个查询中，给定一个节点编号，对于每个查询，按编号从小到大输出该节点所有子节点。

Input
第一行两个整数n, m，之后n行，每行两个整数a, b，表示编号为a的节点的父节点是b，b为0表示没有父节点，数据保证编号为1至n的节点均在a位置出现一次，之后一行m个整数，表示每个查询要查询的节点编号。

Output
m行，每行若干个从小到大排好序的、用一个空格隔开的整数，表示该次查询的节点的所有子节点。

Sample Input
5 1
1 0
4 1
2 1
5 1
3 1
1

Sample Output
2 3 4 5

Source
Unknown

解析一下吧，说实话这是一道很水的题目，前提是你看懂题目了…我一开始把子节点理解成子节点、子节点的子节点、子节点的子节点的子节点…事实证明我想多了，其实就这么简单，不用vector也可以，我用vector只是练练手，大家仅供参考。
附上ac代码：

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#include <map>
#include <queue>
using namespace std;
int c[1000005];
map<int,vector<int> > sh;
int main()
{
    int n,m;
    scanf("%d %d",&n,&m);
    while(n--)
    {
        int a,b;
        scanf("%d %d",&a,&b);
        sh[b].push_back(a);
    }
    while(m--)
    {
        memset(c,0,sizeof(c));
        int a;
        scanf("%d",&a);
        int j=0;
        for(int i=0;i<sh[a].size();i++,j++)
        {
            c[j]=sh[a][i];
        }
        sort(c,c+j);
        if(c[j-1]!=0)
        {
            for(int i=0; i<j-1; i++)
                printf("%d ",c[i]);
            cout<<c[j-1];
        }
        cout<<endl;
    }
    return 0;
}

那么我们索性拓展一下，就以我一开始想错的点出发，写一道题，就是输出子节点之下的所有节点，可以想一想，没有题，附上我的代码给大家参考一下（因为没有题，也不知道ac不。但至少我没有找到错误。）

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#include <map>
#include <queue>
using namespace std;
int c[1000005];
map<int,vector<int> > sh;
int main()
{
    int n,m;
    scanf("%d %d",&n,&m);
    while(n--)
    {
        int a,b;
        scanf("%d %d",&a,&b);
        sh[b].push_back(a);
    }
    while(m--)
    {
        memset(c,0,sizeof(c));
        int a;
        scanf("%d",&a);
        queue<int>js;
        js.push(a);

        int j=0;
        while(!js.empty())
        {
            int b=js.front();
            js.pop();
            for(int i=0; i<sh[b].size(); i++)
            {
                c[j]=sh[b][i];
                j++;
                js.push(sh[b][i]);
            }
        }
        sort(c,c+j);
        if(c[j-1]!=0)
        {
            for(int i=0; i<j-1; i++)
                printf("%d ",c[i]);
            cout<<c[j-1]<<endl;
        }
    }
    return 0;
}

唉，一直觉得自己英语不好，没想到语文也不咋地，多看书啊！！！