【ARC103D】Distance Sums【树论】【构造】
題意:一棵nnn個(gè)點(diǎn)的無(wú)權(quán)樹(shù),給定每個(gè)點(diǎn)iii到其他所有點(diǎn)的距離之和did_idi?,保證did_idi?兩兩不同。構(gòu)造或判斷無(wú)法構(gòu)造一棵滿(mǎn)足條件的樹(shù)。
n≤105n\leq 10^5n≤105
首先對(duì)于非根結(jié)點(diǎn)uuu,有
du=dfau?sizu+(n?sizu)d_u=d_{fa_u}-siz_u+(n-siz_u)du?=dfau???sizu?+(n?sizu?)
所以ddd最大的一定是葉子結(jié)點(diǎn)
把ddd從大到小排序,然后用上面的式子依次找到每個(gè)點(diǎn)的父親,如果沒(méi)有找到輸出無(wú)解。
注意并不是由兒子的ddd來(lái)確定它的父親是誰(shuí),它的父親以及ddd值是早就定好了的,你只是把它找出來(lái)。這個(gè)式子是計(jì)算式而不是決定式。
因?yàn)?span id="ozvdkddzhkzd" class="katex--inline">ddd值是唯一的,所以這個(gè)過(guò)程是確定的,如果有解的話(huà)一定可以確定出來(lái)。
注意構(gòu)造出來(lái)了不一定代表有解,如果把ddd看成nnn個(gè)變量,而n?1n-1n?1條邊只能確定ddd之間的關(guān)系。所以需要判斷一個(gè)點(diǎn)算出的ddd是否是真正的ddd,直接把sizsizsiz加起來(lái)就是drootd_{root}droot?,判一下即可。
復(fù)雜度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; }總結(jié)
以上是生活随笔為你收集整理的【ARC103D】Distance Sums【树论】【构造】的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇: 慢性肝炎能治吗
- 下一篇: 【CF1349D】Slime and B