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;
}
啊哈哈哈,我滴任务完成啦。