zoukankan      html  css  js  c++  java
  • LeetCode-Populating Next Right Pointers in Each Node

    Given a binary tree

        struct TreeLinkNode {
          TreeLinkNode *left;
          TreeLinkNode *right;
          TreeLinkNode *next;
        }
    

    Populate each next pointer to point to its next right node. If there is no next right node, the next pointer should be set to NULL.

    Initially, all next pointers are set to NULL.

    Note:

    • You may only use constant extra space.
    • You may assume that it is a perfect binary tree (ie, all leaves are at the same level, and every parent has two children).

    For example,
    Given the following perfect binary tree,

             1
           /  
          2    3
         /   / 
        4  5  6  7
    

    After calling your function, the tree should look like:

    1 -> NULL
           /  
          2 -> 3 -> NULL
         /   / 
        4->5->6->7 -> NULL
    /**
     * Definition for binary tree with next pointer.
     * struct TreeLinkNode {
     *  int val;
     *  TreeLinkNode *left, *right, *next;
     *  TreeLinkNode(int x) : val(x), left(NULL), right(NULL), next(NULL) {}
     * };
     */
    class Solution {
    public:
        int Count(TreeLinkNode *root){
            if(root==NULL)return 0;
            else return Count(root->left)+Count(root->right)+1;
        }
        void connect(TreeLinkNode *root) {
            // Start typing your C/C++ solution below
            // DO NOT write int main() function
            if(root==NULL)return;
            int count=Count(root);
            vector<TreeLinkNode*> queue;
            queue.resize(count);
            int head=0,tail=1;
            queue[0]=root;
            int current=1;
            int next=0;
            while(head<count){
                
                if(queue[head]->left!=NULL){
                    queue[tail]=queue[head]->left;
                    tail++;
                    next++;
                }
                 if(queue[head]->right!=NULL){
                    queue[tail]=queue[head]->right;
                    tail++;
                    next++;
                }
                current--;
                if(current==0){
                    current=next;
                    next=0;
                }
                else{
                    queue[head]->next=queue[head+1];
                }
                head++;
            }
        }
    };
    
  • 相关阅读:
    Asp.net 文件下载,中文文件名出现乱码
    存储过程中执行动态Sql语句
    SQLServer : EXEC和sp_executesql的区别
    C# 从服务器下载文件代码
    在windows 7 上为 sqlserver 2008 启用远程访问
    数字求和
    大道至简第一章读后感
    作业
    SharePoint2013 App 开发中 自定义网站栏,内容类型,列表。
    SharePoint2013 App开发模式
  • 原文地址:https://www.cnblogs.com/superzrx/p/3326590.html
Copyright © 2011-2022 走看看