第五章 树9 AcWing 1600. 完全二叉树

第五章 树9 AcWing 1600. 完全二叉树

原题链接

AcWing 1600. 完全二叉树

算法标签

树

思路

对于完全二叉树，具有如下性质

因此， 从根节点开始遍历，若为完全二叉树，则恰好可以填入1-n (根节点对应位置是1) 否则最后一个节点对应的位置一定大于n（由于1-n中存在未填充位置）

输入处理

对于N行输入
若用

char a , b;
cin >> a >> b 

会Wrong Answer
输入的数字不仅只有个位数（0-9），可能会有两位数字（大于等于10的数）， char字符无法读取两位数字造成数据丢失

#pragma GCC optimize(2)
#pragma GCC optimize(3)
#include<bits/stdc++.h>
#define int long long
#define x first
#define y second
#define ump unordered_map
#define pq priority_queue
#define rep(i, a, b) for(int i=a;i<b;++i)
#define Rep(i, a, b) for(int i=a;i>=b;--i)
using namespace std;
typedef pair<int, int> PII;
const int N = 25;
//int t, n, m, cnt, ans; 
int l[N], r[N];
bool f[N];
int mxk, mxid;
inline int rd(){
   int s=0,w=1;
   char ch=getchar();
   while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
   while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
   return s*w;
}
void put(int x) {
    if(x<0) putchar('-'),x=-x;
    if(x>=10) put(x/10);
    putchar(x%10^48);
}

void dfs(int u, int k){
    if(u==-1){
        return;
    }
    if(k>mxk){
        mxk=k;
        mxid=u;
    }
    dfs(l[u], k*2);
    dfs(r[u], k*2+1);
}
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	memset(l, -1, sizeof l);
	memset(r, -1, sizeof r);
	int n;
	cin>>n;
	string op1,op2;
	rep(i, 0, n){
	    cin>>op1>>op2;
	    if(op1!="-"){
	        l[i]=stoi(op1);
	        f[l[i]]=true;
	    }
	    if(op2!="-"){
	        r[i]=stoi(op2);
	        f[r[i]]=true;
	    }
	}
	int root=0;
	while(f[root]){
	    root++;
	}
	// 从根节点开始递归 根节点对应数组下标为1
	dfs(root, 1);
	if(mxk==n){
	    cout<<"YES "<<mxid;
	}else{
	    cout<<"NO "<<root;
	}
	return 0;
}

参考文献

AcWing 1600. 完全二叉树（PAT甲级辅导课）

原创不易
转载请标明出处
如果对你有所帮助 别忘啦点赞支持哈