每日一题之 hiho1542 无根树变有根树

最新推荐文章于 2020-02-26 16:30:33 发布

alwaysRememberrr

最新推荐文章于 2020-02-26 16:30:33 发布

阅读量262

点赞数

分类专栏：数据结构&算法

本文链接：https://blog.csdn.net/u014046022/article/details/104203025

版权

数据结构&算法专栏收录该内容

163 篇文章 1 订阅

订阅专栏

这是一篇关于如何将一棵包含N个节点的无根树指定一个节点K作为根，从而确定所有节点父节点的解题文章。输入包括树的节点数N和根节点K，以及N-1条边的连接信息。输出则列出每个节点的父节点编号，根节点K的父节点标记为0。

摘要由CSDN通过智能技术生成

时间限制: 10000ms

单点时限: 1000ms

内存限制: 256MB

描述

给定一棵包含 N 个节点的无根树，小Hi想知道如果指定其中某个节点 K 为根，那么每个节点的父节点是谁？

输入

第一行包含一个整数 N 和 K。1 ≤ N ≤ 1000, 1 ≤ K ≤ N。

以下N-1行每行包含两个整数 a 和 b，代表ab之间存在一条边。 1 ≤ a, b ≤ N。

输入保证是一棵树。

输出

输出一行包含 N 个整数，分别代表1~N的父节点的编号。对于 K 的父节点输出0。

样例输入

样例输出

3 1 4 0 1

思路：

建图后从给定的节点K开始dfs，dfs过程中记录parent节点即可

#include <iostream>
#include <vector>
#include <cstring>

using namespace std;

const int maxn = 1005;

vector<int>G[maxn];
bool vis[maxn];
int parent[maxn];

void dfs(int v) {
	vis[v] = 1;
	for (int i = 0; i < G[v].size(); ++i) {
		auto u = G[v][i];
		if (!vis[u]) {
			dfs(u);
			parent[u] = v;
		}
	}
}

void solve(int &N, int &K) {
	int u,v;
	memset(vis, 0, sizeof(vis));
	memset(parent, 0, sizeof(parent));
	for (int i = 0; i < N-1; ++i) {
		cin >> u >> v;
		G[u].push_back(v);
		G[v].push_back(u);
	}
	parent[K] = 0;
	dfs(K);
	for (int i = 1; i <= N; ++i) {
		cout << parent[i] << " ";
	}

}
int main() {
	int N, K;
	cin >> N >> K;
	solve(N, K);
	return 0;
}