根据《剑指Offer——名企面试官精讲典型编程题》上的讲解,实现代码如下:
#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<vector>
using namespace std;
//二叉树结点
typedef struct BiTNode{
//数据
int data;
//左右孩子指针
struct BiTNode *lchild,*rchild;
}BiTNode,*BiTree;
//按先序序列创建二叉树
int CreateBiTree(BiTree &T){
int data;
//按先序次序输入二叉树中结点的值(一个字符),‘#’表示空树
scanf("%d",&data);
if(data == -1){
T = NULL;
}
else{
T = (BiTree)malloc(sizeof(BiTNode));
//生成根结点
T->data = data;
//构造左子树
CreateBiTree(T->lchild);
//构造右子树
CreateBiTree(T->rchild);
}
return 0;
}
void FindPath(BiTNode* pRoot, int expectedSum, std::vector<int>& path, int& currentSum);
void FindPath(BiTNode* pRoot, int expectedSum)
{
if(pRoot == NULL)
return;
std::vector<int> path;
int currentSum = 0;
FindPath(pRoot, expectedSum, path, currentSum);
}
void FindPath(BiTNode* pRoot, int expectedSum, std::vector<int>& path, int& currentSum)
{
currentSum += pRoot->data;
path.push_back(pRoot->data);
// 如果是叶结点,并且路径上结点的和等于输入的值
// 打印出这条路径
bool isLeaf = pRoot->lchild == NULL && pRoot->rchild == NULL;
if(currentSum == expectedSum && isLeaf)
{
printf("A path is found: ");
std::vector<int>::iterator iter = path.begin();//选择vector而不用stack的原因
for(; iter != path.end(); ++ iter)
printf("%d\t", *iter);
printf("\n");
}
// 如果不是叶结点,则遍历它的子结点
if(pRoot->lchild != NULL)
FindPath(pRoot->lchild, expectedSum, path, currentSum);
if(pRoot->rchild != NULL)
FindPath(pRoot->rchild, expectedSum, path, currentSum);
// 在返回到父结点之前,在路径上删除当前结点,
// 并在currentSum中减去当前结点的值
currentSum -= pRoot->data;
path.pop_back();
}
int main()
{
BiTree T;
CreateBiTree(T);
FindPath(T, 22);
return 0;
}