二叉树和为某一值的路径-路径必须是从根节点出发,还必须非得到达叶节点。
程序员文章站
2022-07-09 12:39:16
...
时间限制:C/C++ 1秒,其他语言2秒 空间限制:C/C++ 32M,其他语言64M 热度指数:589567
本题知识点: 树
题目描述
输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。一看就是后序遍历。
解答:
递归方法:
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
private:
vector<vector<int> >re;
vector<int>temp;
public:
vector<vector<int> > FindPath(TreeNode* root,int n) {
//在 Dev C++ 中,上面写法中 int 后面的两个>之间需要有空格,
//否则有的编译器会把它们当作>>运算符,编译会出错。
if(root==NULL){
return re;
}
int rootnum=root->val;
int sum=n-rootnum;
temp.push_back(rootnum);
FindPath(root->left,sum);
if(root->left!=NULL)temp.pop_back();
FindPath(root->right,sum);
if(root->right!=NULL)temp.pop_back();
if(root->left==NULL&&root->right==NULL){
if(rootnum==n){
re.push_back(temp);
}
}
return re;
}
};
非递归方法:
class Solution {
public:
vector<vector<int> > FindPath(TreeNode* root, int n) {
//在 Dev C++ 中,上面写法中 int 后面的两个>之间需要有空格,
//否则有的编译器会把它们当作>>运算符,编译会出错。
vector<vector<int> >re;
vector<int>temp;
vector<TreeNode*>tempsc;
stack<TreeNode*>sc;
TreeNode*p = root;
TreeNode*pre = NULL;
int sum = 0;
int recount = 0;
while (p != NULL || !sc.empty()){
if (p != NULL){//注意这里是if就可以,用来检测有没有左节点。
sum = sum + p->val;
sc.push(p);
p = p->left;
}
else{
p = sc.top();
p = p->right;
if (p != NULL&&p != pre){//用来检测有没有右节点。
sc.push(p);
sum = sum + p->val;
p = p->left;
}
else{
int topnum = sc.top()->val;
p = sc.top();
//给指针赋值别总忘记。
if (sum == n&&p->right==NULL){
recount++;
while (!sc.empty()){
tempsc.push_back(sc.top());
temp.push_back(sc.top()->val);
sc.pop();
}
reverse(tempsc.begin(), tempsc.end());
reverse(temp.begin(), temp.end());
for (int i = 0; i<tempsc.size(); i++){
sc.push(tempsc[i]);
}
re.push_back(temp);
temp.clear();
tempsc.clear();
}
sc.pop();
sum = sum - topnum;
pre = p;
p = NULL;
//访问完别忘记置空,只有置空才可以知道该节点的左节点访问结束了,形成完整的逻辑链
}
}
}
return re;
}
};
上一篇: 攻防世界 Crypto进阶 简单的rsa