PAT 甲级 1110 Complete Binary Tree (25 分) 完全二叉树 DFS

原题链接：1110 Complete Binary Tree (25 分)

题目大意：

给你一颗树，你来判断一下它是否是一颗完全二叉树。
输入：N个点
接下来N行代表每个点的左右子结点的信息，并给出该结点的左右子结点的编号，如果某个子结点不存在，则用 - 代替。

题目分析：

    我们可以假设这就是一个完全二叉树，建树。将空的地方留下，其他地方慢慢填满。
填完之后对这颗树进行DFS，每个点除了结点值，我们还要为其分配一个从1开始的下标值，因为在完全二叉树中，结点的左子树的下标是 2n，右子树的下标是 2n+1。我们就利用这个性质，对每一个结点的下标进行记录，然后记录最大的一个下标和结点值就行了。
回到主函数中，用记录的最大下标与n进行比较，输出即可。

满分代码：

#include <iostream>
#include <set>
#include <map>
#include <vector>
#include <cmath>
#include <cstring>
#include <algorithm>
#define inf 0x3f3f3f3f
typedef long long LL;
using namespace std;
const int MAXN = 30;
/*
	假设这就是一个完全二叉树，建树。将空的地方留下，其他地方慢慢填满
*/
int n;
int l[MAXN], r[MAXN];
int maxk, maxu;
bool has_father[MAXN];

// 结点编号  结点下标 
void dfs(int u, int k) {
	if(u == -1)	return;
	if(k > maxk) {
		maxk = k;
		maxu = u;
	}
	dfs(l[u], k*2);
	dfs(r[u], k*2+1);
}

int main() {
	cin >> n;
	memset(l, -1, sizeof l);
	memset(r, -1, sizeof r);
	for(int i = 0; i < n; i++) {
		string a, b;
		cin >> a >> b;
		// 如果存在某儿子 将string'转换成int' 并且标记它们存在父结点 
		if(a != "-")	l[i] = stoi(a), has_father[l[i]] = 1;
		if(b != "-")	r[i] = stoi(b), has_father[r[i]] = 1;
	}
	int root;
	// 遍历一次找到没有父结点的就是根结点 
	for(int i = 0; i < n; i++) {
		if(!has_father[i]) {
			root = i;
			break;
		}
	}
	dfs(root, 1);
	if(maxk == n)	printf("YES %d\n", maxu);
	else	printf("NO %d\n", root);
	
	return 0;
}