SDNUOJ——1057.树的查询

最新推荐文章于 2025-07-10 21:57:36 发布

浦柳人

最新推荐文章于 2025-07-10 21:57:36 发布

阅读量294

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/weixin_43237242/article/details/90703073

SDNUOJ 专栏收录该内容

42 篇文章

订阅专栏

博客围绕树节点子节点查询问题展开，给定树中节点个数 n 和查询数量 m，以及各节点父节点编号和查询的节点编号，需按编号从小到大输出查询节点的所有子节点，并给出了输入输出格式及示例。

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

#include<stdio.h>
#include<iostream>
#include<map>
#include<algorithm>
#include<cstring>
#include<string.h>
#include<string>
#include<math.h>
#include<vector>
#include<map>
using namespace std;
typedef long long ll;
#define MAXN 1000005
#define INF 0x3f3f3f3f//将近int类型最大数的一半，而且乘2不会爆int
const ll MOD = 332748118;

vector<int>g[MAXN];
int cf[MAXN];
int main()
{
    int n, m;
    cin >> n >> m;
    for(int i=0; i<n; ++i)
    {
        int a, b;
        scanf("%d %d", &a, &b);
        g[b].push_back(a);
    }
    for(int i=0; i<m; ++i)
    {
        int c;
        scanf("%d", &c);
        int len = g[c].size();
        for(int i=0; i<len; ++i)
            cf[i] = g[c][i];
        sort(cf, cf+len);
        for(int i=0; i<len; ++i)
        {
            cout << cf[i];
            if(i != len-1)
                cout << ' ';
        }
            cout << '\n';
    }
    return 0;
}