题意
给出n和一个长度为n的数组d,满足d数组中的任意两个元素互不相同,要求构造一棵n个节点边权为1的树,使得所有节点到节点i的距离和恰好为d[i]。
n
≤
100000
n\le100000
n≤100000
分析
显然按d[i]排序之后,要么从小到大构造,要么从大到小构造。
由于d[i]互不相同,不难发现d[i]最小的一定是重心,d[i]最大的一定是叶子。
由于一棵树中除了根以外每个节点只有一个父亲,我们考虑从大到小构造,也就是把这棵树的重心作为根后,从底往上构造。
假设当前枚举到点x,x的子树大小为sz,那么x父亲的d就等于d[x]-(n-sz)+sz,这样我们就可以把x的父亲定下来。
整棵树构造完之后,我们还需要对其中某个点进行验证,因为你把所有的d[x]同时加上一个数后你仍然可以构造出这棵树。
然后就做完了。
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<map>
#include<vector>
#define pb push_back
typedef long long LL;
const int N=100005;
int n,size[N],id[N],ans1[N],ans2[N];
LL tot,a[N];
std::map<LL,int> num;
std::vector<int> vec[N];
bool cmp(int x,int y)
{
return a[x]>a[y];
}
void dfs(int x,int fa,int d)
{
tot+=(LL)d;
for (int i=0;i<vec[x].size();i++)
if (vec[x][i]!=fa) dfs(vec[x][i],x,d+1);
}
int main()
{
scanf("%d",&n);
for (int i=1;i<=n;i++) scanf("%lld",&a[i]),num[a[i]]=i;
for (int i=1;i<=n;i++) size[i]=1,id[i]=i;
std::sort(id+1,id+n+1,cmp);
for (int i=1;i<n;i++)
{
int x=id[i];LL w=a[x]-(n-size[x])+size[x];
if (!num[w]||w>=a[x]) {puts("-1");return 0;}
int y=num[w];
ans1[i]=x;ans2[i]=y;
vec[x].pb(y);vec[y].pb(x);
size[y]+=size[x];
}
dfs(id[n],0,0);
if (tot!=a[id[n]]) {puts("-1");return 0;}
for (int i=1;i<n;i++) printf("%d %d\n",ans1[i],ans2[i]);
return 0;
}