链接: D. Dogeforces
题意:
给出一棵树的叶子节点数 n ,和任意两个叶子节点的 lca 的权值。要求构造出这棵树,使得父节点的权值比儿子节点大并且每个出叶子节点以外的节点至少有两个儿子节点。
思路:
因为父节点的权值比儿子节点大,所以可以按照权值排序,就可以并查集模拟从叶子节点往上构造。在每次合并中如果当前两节点的父亲的权值与当前权值都不同,那么就要加入一个新的点当成两个节点的父亲(加入两条边)。否则就让那个权值大的当作父亲(加一条边)。每次合并都要让权值大的当父亲。还有在排序的时候要按照 i 排序 具体可参考 添加链接描述
代码:
#include<iostream>
#include<cstdio>
#include<set>
#include<map>
#include<bitset>
#include<cstring>
#include<algorithm>
#define fin(x) freopen(x, "r", stdin)
#define fout(x) freopen(x, "w", stdout)
using namespace std;
typedef long long ll;
const int maxn=1e6+7;
int n,a[600][600],fa[maxn],val[maxn];
struct node{
int u,v,val;
}e[maxn];
bool cmp(node a,node b){
if(a.val == b.val){
return a.u < b.u;
}
return a.val < b.val;
}
set<pair<int , int>> ans;
int find(int x){
if(x != fa[x]) fa[x] = find(fa[x]);
return fa[x];
}
void conbine(int u,int v){
int temp1 = find(u);
int temp2 = find(v);
if(temp1 < temp2) swap(temp1,temp2);
if(temp1 != temp2){
fa[temp2] = temp1;
}
}
int main(){
scanf("%d",&n);
int cnt = 0;
for(int i = 1; i <= n * n; i ++){
fa[i] = i;
}
for(int i = 1; i <= n ; i ++){
for(int j = 1; j <= n; j ++){
scanf("%d",&a[i][j]);
if(i < j) e[++cnt] = {i , j , a[i][j]};
if(i == j) val[i] = a[i][j];
}
}
sort(e + 1,e + cnt + 1,cmp);
for(int i = 1; i <= cnt; i ++){
int u = e[i].u,v = e[i].v,w = e[i].val;
u = find(u),v = find(v);
if(u == v) continue;
if(u > v) swap(u,v);
if(val[u] == w || val[v] == w){
conbine(u,v);
ans.insert({u,v});
}
else{
val[++n] = w;
conbine(n,u);
conbine(n,v);
ans.insert({u,n});
ans.insert({v,n});
}
}
printf ("%d\n",n);
for(int i = 1; i <= n; i ++){
printf ("%d ",val[i]);
}
printf ("\n%d\n",n);
for(auto p : ans){
printf ("%d %d\n",p.first,p.second);
}
}