题目:
计算一棵给定二叉树的所有双分支结点个数。
递归:
思路:
- 定义一个参数 int ans = 0;
- 若当前结点有左孩子和右孩子则ans+1;
- 若存在左孩子 则左孩子执行 1,2步骤;若存在右孩子则右孩子执行1,2步骤;
测试用例:
ABD##E##DF##G## 结果:3
ABD###CE##F## 结果:2
代码实现:
#include <iostream>
using namespace std;
#define Max 20
typedef struct treenode{
char data;
struct treenode *lchild,*rchild;
}treenode,*tree;
void buildtree(tree &t){
char ch;
ch = getchar();
if (ch=='#') t = NULL;
else{
t = (treenode *)malloc(sizeof(treenode));
t->data = ch;
t->lchild = NULL;
t->rchild = NULL;
buildtree(t->lchild);
buildtree(t->rchild);
}
}
int getdigui(tree t,int &ans){
if(t->lchild!=NULL&&t->rchild!=NULL) ans++;
if(t->lchild) getdigui(t->lchild,ans);
if(t->rchild) getdigui(t->rchild,ans);
return ans;
}
int main(){
tree t;
int ans;
buildtree(t);
getdigui(t,ans);
cout<<ans;
return 0;
}
非递归:
思路:
- 辅助队列q(用来层序遍历二叉树), int ans =0(用于记录结果值);
- 若结点不为空,则结点入队
- 若队列不为空,则出队,并判断出队结点是否同时有左右孩子,若有则ans+1;
- 若左孩子存在则左孩子入队,若右孩子存在则右孩子入队。
- 重复2、3、4步骤;
测试用例:
ABD##E##DF##G## 结果:3
ABD###CE##F## 结果:2
代码示例:
#include <iostream>
using namespace std;
#define Max 20
typedef struct treenode{
char data;
struct treenode *lchild,*rchild;
}treenode,*tree;
void buildtree(tree &t){
char ch;
ch = getchar();
if (ch=='#') t = NULL;
else{
t = (treenode *)malloc(sizeof(treenode));
t->data = ch;
t->lchild = NULL;
t->rchild = NULL;
buildtree(t->lchild);
buildtree(t->rchild);
}
}
struct squeue{
struct treenode *data[Max];
int f,r,tag;
};
bool isEmpty(squeue q){
if(q.f == q.r&&q.tag==0){
return true;
}
return false;
}
bool isFull(squeue q){
if (q.f==q.r&& q.tag==1)
{
return true;
}
return false;
}
//入队
bool enter(squeue &q,tree t){
if (isFull(q))
{
return false;
}
q.data[q.r] = t;
q.r = (q.r+1)%Max;
q.tag =1;
return true;
}
//出队
bool outs(squeue &q,treenode *&p){
if (isEmpty(q))
{
return false;
}
p = q.data[q.f];
q.f = (q.f+1)%Max;
q.tag = 0;
return true;
}
int getTwoTree(tree t){
int ans=0;
squeue q;
q.f = q.r = q.tag = 0;
if (t==NULL) return 0;
enter(q,t);
treenode *p;
while (q.f<q.r)
{
outs(q,p);
if(p->lchild!=NULL&&p->rchild) ans++;
if(p->lchild!=NULL) enter(q,p->lchild);
if(p->rchild!=NULL) enter(q,p->rchild);
}
cout<<"双分支结点个数为:"<<ans<<endl;
return ans;
}
int main(){
tree t;
buildtree(t);
getTwoTree(t);
return 0;
}