码迷,mamicode.com
首页 > 其他好文 > 详细

Binary Tree Paths

时间:2017-06-27 14:55:23      阅读:151      评论:0      收藏:0      [点我收藏+]

标签:vector   ini   return   rac   treenode   tor   following   pre   roo   

Given a binary tree, return all root-to-leaf paths.

For example, given the following binary tree:

   1
 /   2     3
   5

All root-to-leaf paths are:

["1->2->5", "1->3"]


/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<string>ans;
    vector<string> binaryTreePaths(TreeNode* root) {
        if(root == NULL) return ans;
        string head = to_string(root->val);
        if(root->left) oneTreePath(head, root->left);
        if(root->right) oneTreePath(head, root->right);
        if(!root->left && !root->right)
        ans.push_back(head);
        return ans;
    }
    void oneTreePath(string head, TreeNode* root){
      string newHead = head + "->" + to_string(root->val);
      if(root->left) oneTreePath(newHead, root->left);
      if(root->right) oneTreePath(newHead, root->right);
      if(!root->left && !root->right)
        ans.push_back(newHead); 
    }
};


Binary Tree Paths

标签:vector   ini   return   rac   treenode   tor   following   pre   roo   

原文地址:http://www.cnblogs.com/jzssuanfa/p/7084712.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!