題意 ：

• 給定n個結點的樹和序列$b$和$p$，$b_i$表示i結點的父節點，其中$b_{root}=root$，現在要給樹上的每個邊賦正權值，使得每個結點到根的距離$dis{t}_i$滿足p的排序，即如果在p數組中，i點位置在j點前面，則$dis{t}_i<dis{t}_j$
• 如果可行，輸出每條邊的權值，否則，輸出-1

思路 ：

• 因為每條邊都賦正值，那么考慮一個結點，它的排名不可能比父節點靠前，也就是說p數組滿足拓撲序
• 對于$p=[3,1,2,5,4]$，構造$dis{t}_3=0$（第一個肯定是根結點），$dis{t}_1=2,dis{t}_2=3,...,$滿足枚舉的順序，這個是結點到根的距離，只要記錄一下父節點到根的距離，相減就是這個邊的權值
• 因此，按照p數組的順序從第一個非根結點開始枚舉每個點，如果枚舉到的這個點的父節點排名比它低（dist數組值為-1），直接return輸出-1，否則，當前這個結點dist數組值為i，ans數組記錄這個點到父節點這條邊的邊權，就是當前這個點到根的距離減去父節點到根的距離

#include <iostream> #include <algorithm> #include <cstring> #include <vector> #include <unordered_set> #include <math.h> using namespace std;typedef long long ll; typedef pair<int, int> PII;#define endl '\n' #define fi first #define se second #define push_back #define rep(i, l, r) for (ll i = l; i <= r; i ++ )const int N = 2e5 + 10;int b[N], p[N]; int ans[N], dist[N]; // ans為i點到父節點的距離，即邊權；dist為i點到根結點的距離void solve() {int n; cin >> n;for (int i = 1; i <= n && cin >> b[i]; i ++ );for (int i = 1; i <= n && cin >> p[i]; i ++ );for (int i = 1; i <= n; i ++ ) ans[i] = dist[i] = -1; // 初始化！同時也可以標記該點是否已經被訪問過，相對排名ans[p[1]] = 0, dist[p[1]] = 0;for (int i = 2; i <= n; i ++ ){int j = p[i]; // 當前這個點if (dist[b[j]] == -1) return cout << -1 << endl, void();dist[j] = i;ans[j] = dist[j] - dist[b[j]];}for (int i = 1; i <= n; i ++ ) cout << ans[i] << " \n"[i == n]; }int main() {cin.tie(nullptr) -> sync_with_stdio(false);int _;cin >> _;while (_ -- )solve();return 0; } 與50位技術專家面對面20年技術見證，附贈技術全景圖

總結

以上是生活随笔為你收集整理的Weights Assignment For Tree Edges 树，拓扑序（1500）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。