codeforces 1283F. DIY Garland(树+优先队列)

题目连接:https://codeforces.com/contest/1283/problem/F

题意:一根电线连接着两个点,这两个点分别代表着两个灯,灯有自己的编号i,其亮度是2 ^ i,每根电线的两个灯分别为主灯和副灯,电源从主灯来,电流向副灯。最开始有一个源点灯,所有的电源从此处流入,对于每根电线,定义其重要性为 :切断这根电线后不能通电的所有灯的亮度之和。首先题目按电线的重要性给出n-1条电线的主灯编号,让你从大到小输出每条电线所连接的两个灯的序号(无前后差别)

题解:首先,题目的所描述的结构是一棵树,因为题目是按电线重要性给出的节点主灯,那么第一个节点必定是源点,因为重要性最大。其次因为树的叶子节点度必为1,而且叶子节点必定不是主灯,那么我们可以在输入之后统计叶子节点,同时统计一下输入节点的度。叶子节点的编号用优先队列存储,因为亮度最小最先和重要性低的电线匹配。因为输入的时候电线是按重要性从大到小输入的,那么我们就可以把优先队列中的节点一个一个地与其父亲节点(主灯)匹配,匹配过后,让当前这个主灯的度-1,如果这个此时度变为1,那么加入到优先队列之中,因为他现在成了“叶子节点”,如此过程直到把n-1条边都找到,break即可

AC代码:

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<algorithm>
 5 #include<vector>
 6 #include<queue> 
 7 using namespace std;
 8 const int maxn = 2e5+5;
 9 int ind[maxn];
10 vector<pair<int,int> > ans;
11 int main(){
12     int n;
13     cin>>n;
14     priority_queue<int, vector<int>,greater<int> > q;
15     vector<int> mainLamp;
16     int gird = -1;
17     for(int i = 0;i<n-1;i++){
18         int lamp;
19         cin>>lamp;
20         if(gird == -1) gird = lamp;//记录源点
21         ind[lamp]++;//记录度
22         mainLamp.push_back(lamp);
23     }
24     for(int i = 1;i<=n;i++){
25         if(ind[i] == 0){//寻找未出现的叶子节点
26             q.push(i); 
27         }
28     }
29     while(!q.empty() ){    
30         int cur = q.top();
31         q.pop();
32         int curMainLamp = mainLamp[mainLamp.size()-1];
33         mainLamp.pop_back();//匹配一个主灯,就弹出一个
34         ans.push_back({curMainLamp,cur});//加入一条边
35         if(ans.size() == n -1 ) break;
36         ind[curMainLamp]--;
37         if(ind[curMainLamp] == 0){
38             q.push(curMainLamp);//如果度为0,那么变成了叶子节点,加入队列中
39         }
40     }
41     cout<<gird<<endl;
42     for(int i = ans.size() -1;i>=0 ;i--){
43         cout<<ans[i].first<<" "<<ans[i].second<<endl;
44     }
45     return 0;
46 }
posted @ 2020-01-01 20:48  AaronChang  阅读(495)  评论(0编辑  收藏  举报