P10678 『STA - R6』月 题解
题目简述:
给定一棵 n n n 个节点的树,每个节点的度数 d i d_i di,请你构造一棵树,使数的深度尽可能小。
大致思路:
显而易见的,根据类似贪心的思路,显然要先处理度数大的,让他们尽可能向上靠,也就是用一个结构体存,接着,根据 d i d_i di 从大到小进行排序,进行遍历,用广度优先搜索,将边依依填如,在填中,输出相应连接边的点。要注意的是,由于根没有父节点,我们是从上往下搭建的,因此,在将它放入时,要将它深度加上 1 1 1,以此类推,根据它的深度,来判断插入边的数量,总体复杂度 O ( n log n ) O(n \log n) O(nlogn),可以通过。
代码实现:
#include<bits/stdc++.h>
using namespace std;
const int N = 3e5 + 10;
struct node
{
int d, id, cnt;
}a[N];
int T, n;
bool cmp(node l, node r)
{
return l.d > r.d;
}
void bfs()
{
queue < node > q;
a[1].d ++;//根节点无父亲
int tp = 1;
q.push(a[1]);//将认定的根放进去
while(q.size())
{
node x = q.front();
q.pop();
for(int i = 1;i < x.d; ++ i)
{
q.push(a[++ tp]);
cout << x.id << " " << a[tp].id << "\n";
}
}
}
signed main(){
cin >> T;
while (T --)
{
cin >> n;
for (int i = 1;i <= n; ++ i)
{
cin >> a[i].d;
a[i].id = i;
}
sort(a + 1, a + n + 1, cmp);
bfs();
}
return 0;
}
这样这道题就完成啦!!!