从根到叶的二进制数之和

给出一棵二叉树，其上每个结点的值都是 0 或 1 。每一条从根到叶的路径都代表一个从最高有效位开始的二进制数。例如，如果路径为 0 -> 1 -> 1 -> 0 -> 1，那么它表示二进制数 01101，也就是 13 。

对树上的每一片叶子，我们都要找出从根到该叶子的路径所表示的数字。

以 10^9 + 7 为模，返回这些数字之和。

 

示例：

输入：[1,0,1,0,1,0,1]
输出：22
解释：(100) + (101) + (110) + (111) = 4 + 5 + 6 + 7 = 22
 

提示：

树中的结点数介于 1 和 1000 之间。
node.val 为 0 或 1 。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sum-of-root-to-leaf-binary-numbers
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

/**
 * 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:
    int sumRootToLeaf(TreeNode* root) {
        if(root == NULL)
        {
            return 0;
        }
        TreeNode* p = root;
        TreeNode* r;
        int sum = 0;
        bool flag;
        stack<TreeNode*> s1;
        stack<TreeNode*> s2;
        do
        {
            while(p != NULL)
            {
                s1.push(p);
                s2.push(p);
                p = p->left;
            }  //找到最左下角的节点,这个节点没有左子树
           
            r = NULL;
            flag = true;
            //cout<<"s1.size() = "<<s1.size()<<endl;
            //cout<<"s2.size() = "<<s2.size()<<endl;
            
            while(!s1.empty() && flag)
            {
                p = s1.top();
                if(p->right == r)
                {
                    if(p->left == NULL && p->right == NULL)     //叶子节点
                    {
                        //string tmp;
                        int numtmp = 0;
                        for(int i = 0; i<s1.size(); i++)
                        {
                            //tmp = tmp + s2.pop->val;
                            TreeNode* tmp = s2.top();
                            
                            numtmp = numtmp + (tmp->val)*pow(2,i);
                            //cout<<"numtmp = "<<numtmp<<endl; 
                            s2.pop();
                        }
                        
                        sum = sum + numtmp;
                        //cout<<"sum = "<<sum<<endl;
                        
                    }
                    s2 =s1;
                    p = s1.top();
                    r = p;
                    s1.pop();
                    s2.pop();
                    
                    
                }
                else
                {
                     p = p->right;
                     flag = false;
                }
                
            }
            // return 1;
            
        }while(!s1.empty());
        
        return sum;
    }
};