题目链接:http://codeforces.com/contest/746/problem/G
mamaya,不知道YY了一个什么做法就这样过去了啊 2333
首先我显然可以随便构造出一棵树满足他所给出的深度要求。
但是他还对于叶子节点的数目有要求。
考虑首先构造一棵树最大化在满足给出的深度条件下最大化叶子节点的个数。
显然对于每一层的节点让它们的父亲都指向上一层的同一个点的话就会有最多的叶子节点。
好,接下来考虑如何减少叶子结点。
我就是随便贪心搞的(也许可以被叉?)
按照深度从小到大枚举所有的点,如果这个点$x$是叶子节点,找到他的上一层是某一个也是叶子节点的点$y$,并将$x$的父亲修改为$y$,同时还要注意到修改父亲之后,$x$原本的父亲也可能再度变回叶子节点。这些东西都用一个$vector$来维护就可以了。
复杂度:${O(n)}$
1 #include<iostream> 2 #include<cstdio> 3 #include<algorithm> 4 #include<vector> 5 #include<cstdlib> 6 #include<cmath> 7 #include<cstring> 8 using namespace std; 9 #define maxn 1001000 10 #define llg int 11 #define yyj(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout); 12 llg n,m,deep[maxn],c[maxn],a[maxn],du[maxn],dad[maxn],t,k,db[maxn],tot; 13 vector<llg>d[maxn]; 14 15 inline int getint() 16 { 17 int w=0,q=0; char c=getchar(); 18 while((c<'0' || c>'9') && c!='-') c=getchar(); if(c=='-') q=1,c=getchar(); 19 while (c>='0' && c<='9') w=w*10+c-'0', c=getchar(); return q ? -w : w; 20 } 21 22 int main() 23 { 24 yyj("G"); 25 cin>>n>>t>>k; 26 for (llg i=1;i<=t;i++) a[i]=getint(),c[i]=a[i]; 27 llg p=1; 28 db[0]=1; 29 for (llg i=2;i<=n;i++) 30 { 31 if (c[p]==0) p++; 32 dad[i]=db[p-1]; deep[i]=p; 33 du[db[p-1]]++; 34 if (!db[p]) db[p]=i; 35 c[p]--; 36 } 37 for (llg i=1;i<=n;i++) if (!du[i]) {d[deep[i]].push_back(i); tot++;} 38 if (tot<k) {cout<<-1; return 0;} 39 for (llg i=1;i<=n;i++) 40 if (!du[i]) 41 { 42 if (tot==k) break; 43 if (d[deep[i]-1].size()!=0) 44 { 45 du[dad[i]]--; 46 if (du[dad[i]]==0){ tot++; d[deep[dad[i]]].push_back(dad[i]);} 47 du[d[deep[i]-1][d[deep[i]-1].size()-1]]++; 48 tot--; 49 dad[i]=d[deep[i]-1][d[deep[i]-1].size()-1]; 50 d[deep[i]-1].pop_back(); 51 } 52 } 53 if (tot!=k) {cout<<-1; return 0;} 54 cout<<n<<endl; 55 for (llg i=2;i<=n;i++) printf("%d %d ",dad[i],i); 56 return 0; 57 }