二元树中找出和为某一值的所有路径

来源:互联网 发布:淘宝购物截图生成器 编辑:程序博客网 时间:2024/06/01 07:14
在二元树中找出和为某一值的所有路径(树)
题目:输入一个整数和一棵二元树。
从树的根结点开始往下访问一直到叶结点所经过的所有结点形成一条路径。
打印出和与输入整数相等的所有路径。
例如 输入整数22和如下二元树
  10  
  / \   
 5  12   
 /\   
4  7
则打印出两条路径:10, 12和10, 5, 7。
二元树节点的数据结构定义为:
struct BinaryTreeNode // a node in the binary tree
{
int m_nValue; // value of node
BinaryTreeNode *m_pLeft; // left child of node
BinaryTreeNode *m_pRight; // right child of node

};


//coder:LEE
//20120227
#include<iostream>
#include<vector>
#include<CASSERT>
using namespace std;
struct BinaryTreeNode 
{
int m_nValue;
BinaryTreeNode* m_pLeft;
BinaryTreeNode* m_pRight;
};
void AddNode(BinaryTreeNode*& root,int n)
{
if (!root)
{
root=(BinaryTreeNode*)malloc(sizeof(BinaryTreeNode));
root->m_nValue=n;
root->m_pLeft=NULL;
root->m_pRight=NULL;
return;
}
if(n<root->m_nValue)
AddNode(root->m_pLeft,n);
else
AddNode(root->m_pRight,n);
}
void Traverse(BinaryTreeNode* root)
{
if(!root)
return;
cout<<root->m_nValue<<"  ";
Traverse((root->m_pLeft));
Traverse(root->m_pRight);
}
void PrintRoute(BinaryTreeNode* root,vector<int>& r,int& sum,const int& n)
{
if(!root)
return;
r.push_back(root->m_nValue);
sum+=root->m_nValue;
if (root->m_pLeft==NULL&&root->m_pRight==NULL&&sum==n)
{
// for(int i=0;i<r.size();i++)
// cout<<r[i]<<"  ";
vector<int>::iterator iter=r.begin();
for(;iter!=r.end();++iter)
cout<<*iter<<"  ";
cout<<endl;
// return;此处不能直接return,因为还需要处理pop、sum
}
PrintRoute(root->m_pLeft,r,sum,n);
PrintRoute(root->m_pRight,r,sum,n);
r.pop_back();
sum-=root->m_nValue;


}
int main()
{
BinaryTreeNode* root=NULL;
AddNode(root,10);
AddNode(root,5);
AddNode(root,4);
AddNode(root,12);
AddNode(root,7);
Traverse(root);
cout<<endl;
cout<<"route:"<<endl;
vector<int> r;
int sum=0;//currentSum    
int n=22;//expectedSum
PrintRoute(root,r,sum,n);
return 0;
}

原创粉丝点击