【ARC103D】Distance Sums【树论】【构造】
生活随笔
收集整理的這篇文章主要介紹了
【ARC103D】Distance Sums【树论】【构造】
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
題意:一棵nnn個點的無權樹,給定每個點iii到其他所有點的距離之和did_idi?,保證did_idi?兩兩不同。構造或判斷無法構造一棵滿足條件的樹。
n≤105n\leq 10^5n≤105
首先對于非根結點uuu,有
du=dfau?sizu+(n?sizu)d_u=d_{fa_u}-siz_u+(n-siz_u)du?=dfau???sizu?+(n?sizu?)
所以ddd最大的一定是葉子結點
把ddd從大到小排序,然后用上面的式子依次找到每個點的父親,如果沒有找到輸出無解。
注意并不是由兒子的ddd來確定它的父親是誰,它的父親以及ddd值是早就定好了的,你只是把它找出來。這個式子是計算式而不是決定式。
因為ddd值是唯一的,所以這個過程是確定的,如果有解的話一定可以確定出來。
注意構造出來了不一定代表有解,如果把ddd看成nnn個變量,而n?1n-1n?1條邊只能確定ddd之間的關系。所以需要判斷一個點算出的ddd是否是真正的ddd,直接把sizsizsiz加起來就是drootd_{root}droot?,判一下即可。
復雜度O(nlog?n)O(n\log n)O(nlogn)
#include <cstdio> #include <cstring> #include <cctype> #include <algorithm> #include <utility> #define MAXN 100005 using namespace std; typedef long long ll; typedef pair<ll,int> pi; ll read() {ll ans=0;char c=getchar();while (!isdigit(c)) c=getchar();while (isdigit(c)) ans=(ans<<3)+(ans<<1)+(c^48),c=getchar();return ans; } pi p[MAXN]; int siz[MAXN],fa[MAXN]; ll d[MAXN]; int main() {int n=read();for (int i=1;i<=n;i++) p[i]=make_pair(read(),i),siz[i]=1;sort(p+1,p+n+1);ll sum=0;for (int i=n;i>1;i--){ll t=p[i].first-n+2*siz[p[i].second];int k=lower_bound(p+1,p+n+1,make_pair(t,0))-p;if (p[k].first!=t) return puts("-1"),0;fa[p[i].second]=p[k].second,siz[p[k].second]+=siz[p[i].second];sum+=siz[p[i].second];}if (sum!=p[1].first) return puts("-1"),0;for (int i=1;i<=n;i++) if (fa[i]) printf("%d %d\n",fa[i],i);return 0; }總結
以上是生活随笔為你收集整理的【ARC103D】Distance Sums【树论】【构造】的全部內容,希望文章能夠幫你解決所遇到的問題。