F. Weights Assignment For Tree Edges 题解报告

F. Weights Assignment For Tree Edges 题解报告

标签：图论，无向图

题目链接

题意：

本题的题面相当长，看了半小时才看明白， 最重要的还是耐心。

题目首先给出一个数列b计入父节点，如b[1]=2,说明节点2是节点1的父亲节点，b[2]=2,节点2是2的父亲节点，也就是说节点2是树的根节点。

然后给出一个p节点给每一个节点排序，p[1]=3即表示节点3排第一位。

用dist数组计入每个点到根节点的距离，随后让我们给边权赋值，保证dist[pi]<dist[pi+1]，即排名越后的点到根节点的距离越远。

解题思路

搞懂题意后就能发现题目并没有那么困难。

对于给定的排列p, 如果存在某个节点的父节点比该节点排名靠后, dist[父节点]>dist[该节点]，显然不合法。

我们用ans数组来计入边权。为满足题目条件dist[pi]<dist[pi+1]，可自行设定

dist[pi]+1=dist[pi+1]；

那么可得关键代码求出边权。

ans[p[i]] = dist[p[i - 1]] - dist[b[p[i]]] + 1;
dist[p[i]] = dist[b[p[i]]] + ans[p[i]];

代码实现

#include <iostream>
using namespace std;
const int N = 2 * 1e5 + 50;

int b[N], p[N], dist[N], ans[N];

int main() {
	int t;
	cin >> t;
	while (t--) {
		int n;
		cin >> n;
		int fa = 0;
		for (int i = 1; i <= n; i++) {
			cin >> b[i];
			if (b[i] == i) fa = i;
		}
		for (int i = 1; i <= n; i++)
			cin >> p[i];

		for (int i = 1; i <= n; i++)
			ans[i] = dist[i] = -1;//将数组标记为-1，表示还未标记

		int f = 1;
		if (p[1] != fa) f = 0;
		else
		{   
		ans[p[1]] = 0, dist[p[1]] = 0;
			for (int i=2;i<=n;i++)
			{
				if (dist[b[p[i]]]==-1) { f = 0; break; }
                //若父亲节点还未赋值，说明父亲节点在该节点后，数组不合法
				ans[p[i]] = dist[p[i - 1]] - dist[b[p[i]]] + 1;
				dist[p[i]] = dist[b[p[i]]] + ans[p[i]];
			}
		}
		if (f == 0) cout << -1 << endl;
		else {
			for (int i=1;i<=n;i++) cout << ans[i] << " "; 
			cout << endl;
		}
	}
	return 0;
}

啊哈哈哈，我滴任务完成啦。