记录每个节点的出度,叶子节点出度为0,每删掉一个叶子,度数-1,如果一个节点的出度变成0,那么它变成新的叶子。
先把所有叶子放到优先队列中。
从左往右遍历给定序列,对于root[i],每次取出叶子中编号最小的那个与root[i]相连,并且--degree[ root[i] ],如果degree[ root[i] ]为0,那么把root[i]放入优先队列。
1 #include <cstdio> 2 #include <cstring> 3 #include <cstdlib> 4 #include <algorithm> 5 #include <vector> 6 #include <queue> 7 8 using namespace std; 9 10 const int MAXN = 7510; 11 12 struct LLL 13 { 14 int lf; 15 LLL( int nn = 0 ): lf(nn) { } 16 bool operator<( const LLL& rhs ) const 17 { 18 return lf > rhs.lf; 19 } 20 }; 21 22 int vis[MAXN]; 23 vector<int> Tr[MAXN]; 24 int num[MAXN]; 25 priority_queue<LLL> leaf; 26 27 int main() 28 { 29 int cnt = 0; 30 int N = 0; 31 while ( scanf( "%d", &num[cnt] ) != EOF ) 32 { 33 N = max( N, num[cnt] ); 34 ++cnt; 35 } 36 37 memset( vis, 0, sizeof(vis) ); 38 for ( int i = 0; i < cnt; ++i ) 39 ++vis[ num[i] ]; 40 41 for ( int i = 1; i <= N; ++i ) 42 if ( vis[i] == 0 ) leaf.push( LLL(i) ); 43 44 for ( int i = 1; i <= cnt; ++i ) Tr[i].clear(); 45 46 int i, j; 47 for ( i = 0; i < cnt; ++i ) 48 { 49 LLL tmp = leaf.top(); 50 leaf.pop(); 51 Tr[ num[i] ].push_back( tmp.lf ); 52 Tr[ tmp.lf ].push_back( num[i] ); 53 --vis[ num[i] ]; 54 if ( vis[ num[i] ] == 0 ) 55 leaf.push( LLL( num[i] ) ); 56 } 57 58 for ( i = 1; i <= N; ++i ) 59 { 60 printf( "%d:", i ); 61 sort( Tr[i].begin(), Tr[i].end() ); 62 int sz = Tr[i].size(); 63 for ( j = 0; j < sz; ++j ) 64 printf(" %d", Tr[i][j] ); 65 puts(""); 66 67 } 68 return 0; 69 }