题目大意:
判断是不是完全二叉树,是则输出最后一个节点的下标,否则输出根节点下标。
解题思路:
一开始从输入上投机(只有右子没有左子肯定不是,只有左子没有右子但是数量超过2也不是),只拿到18分(忽略了右子树饱满没有左子树的情况)。正确的做法是维护一个下标参数(指完全二叉树结点与位置对应关系),DFS遍历树,如果最大结点的编号正好等于最大的下标证明是一个完全二叉树,否则必然有空缺的位置(该下标下没有元素)。
代码如下:
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n;
struct node
{
int lchild=-1,rchild=-1;
}nod[30];
int getnum(string s)//字符串转整数
{
int sum=0;
for(int i=0;i<s.length();i++)
{
sum=sum*10+(s[i]-'0');
}
return sum;
}
int ma=-1,tail;
void dfs(int root,int inde)
{
if(inde>ma)//更新最大的下标
{
ma=inde;
tail=root;
}
if(nod[root].lchild!=-1)dfs(nod[root].lchild,inde*2);//左儿子下标理论上就是2*inde
if(nod[root].rchild!=-1)dfs(nod[root].rchild,inde*2+1);
}
int main()
{
int book[30]={0};
cin>>n;
for(int i=1;i<=n;i++)//我是从1开始的,方便算下标
{
string l,r;
cin>>l>>r;
if(l!="-")
{
int num=getnum(l)+1;
nod[i].lchild=num;
book[num]=1;
}
if(r!="-")
{
int num=getnum(r)+1;
nod[i].rchild=num;
book[num]=1;
}
}
int root;
for(int i=1;i<=n;i++)//找根节点
{
if(!book[i])
{
root=i;
break;
}
}
dfs(root,1);
if(ma==n)cout<<"YES "<<tail-1<<endl;
else cout<<"NO "<<root-1<<endl;
return 0;
}