CodeForces - 902C Hashing Trees 构造

http://codeforces.com/problemset/problem/902/D

题意：给出一颗n层的树。以及从第0层开始给出每层的节点个数。求是否存在同构，若存在输出其中任意两种。

题解：可以发现若存在同构：

①该节点上一层的节点数大于1，

②该节点所在层节点个数大于1。

剩下的通过构造即可，代码的root变量是父节点。

代码：

#include<bits/stdc++.h>
#define debug cout<<"aaa"<<endl
#define d(a) cout<<a<<endl
#define pb push_back
#define mem(a,b) memset(a,b,sizeof(a))
#define LL long long
#define lson l,mid,root<<1
#define rson mid+1,r,root<<1|1
#define MIN_INT (-2147483647-1)
#define MAX_INT 2147483647
#define MAX_LL 9223372036854775807i64
#define MIN_LL (-9223372036854775807i64-1)
using namespace std;

const int N = 100000 + 5;
const int M = N * N + 5;
const int mod = 1000000000 + 7;
const int INF = 0x3f3f3f3f;
const double eps = 1e-8;

int main(){
	int n,num[N],ans[N],root=0;
	bool flag=1;
	num[0]=1;
	cin>>n;
	for(int i=0;i<=n;i++){
		cin>>num[i];
		if(i==0) continue;
		if(num[i-1]!=1&&num[i]!=1){
			flag=0;
		}
	}
	if(flag){
		puts("perfect");
		return 0;
	}
	puts("ambiguous");
	for(int i=0;i<=n;i++){
		for(int j=1;j<=num[i];j++){
			cout<<root<<" ";
		}
		//现在这个root就是下一层的最后一个节点编号了 
		root+=num[i];
	}
	cout<<endl;
	root=0;
	for(int i=0;i<=n;i++){
		if(i!=0&&num[i-1]!=1&&num[i]!=1){
			for(int j=1;j<=num[i]-1;j++){
				cout<<root<<" ";
			}
			//因为要同构所以构造时，保留一个输出不一样的节点即可 
			cout<<root-1<<" ";
			root+=num[i];
			continue; 
		}
		for(int j=1;j<=num[i];j++){
			cout<<root<<" ";
		}
		root+=num[i];
	}
	return 0;
}