【LeetCode】二叉树OJ

目录

一、根据二叉树创建字符串

二、二叉树的层序遍历

三、二叉树的层序遍历 II

四、二叉树的最近公共祖先

五、二叉搜索树与双向链表

六、从前序与中序遍历序列构造二叉树

七、从中序与后序遍历序列构造二叉树


一、根据二叉树创建字符串

606. 根据二叉树创建字符串 - 力扣(LeetCode)

解题思路:本题在递归前序遍历二叉树的同时要注意()的省略,这时就有了三种情况:

  • 如果当前节点没有孩子,我们不需要在节点后面加上任何括号

  • 如果当前节点只有左孩子,我们在递归时,只需要在左孩子的结果外加上一层括号,而不需要给右孩子加上任何括号
  • 如果当前节点只有右孩子,我们在递归时,需要先加上一层空的括号 ‘()’ 表示左孩子为空,再对右孩子进行递归,并在结果外加上一层括号

解题代码: 

class Solution {
public:
    string tree2str(TreeNode* root) {
        if(root==nullptr)
            return "";
        string str=to_string(root->val);
        if(root->left||root->right)//有左孩子或者无左孩子有右孩子
        {
            str+="(";
            str+=tree2str(root->left);
            str+=")";
        }
       if(root->right)//有右孩子
        {
            str+="(";
            str+=tree2str(root->right);
            str+=")";
        }
        return str;
    }
};

二、二叉树的层序遍历

102. 二叉树的层序遍历 - 力扣(LeetCode)

解题思路:本题为了达到层序遍历,我们需要用到一个队列来保存每一层的节点,将将根节点入队列,每次出队列时都要将其孩子节点入队列:【LeetCode】二叉树OJ_第1张图片

还要有一个变量来记录每一层节点的个数,当该变量不为0时需要将节点出队列,所出节点都属于同一层的元素,每出一个节点将其减1。当该变量为0时,队列中都是上一层节点的孩子节点,所以都属于同一层节点,这时将变量置为队列中元素个数继续将节点出队列即可,直到队列为空

解题代码:

class Solution {
public:
    vector> levelOrder(TreeNode* root) {
        queue q;
        int levelsize=0;
        if(root)//根节点入队列
        {
            q.push(root);
            levelsize=1;
        }
        vector> vv;
        while(!q.empty())
        {
            vector v;//记录每一层的数据
            while(levelsize--)//将上一层节点出队列的同时,将其孩子节点都入队列
            {
                TreeNode* front=q.front();
                q.pop();
                v.push_back(front->val);
                if(front->left)
                    q.push(front->left);
                if(front->right)
                    q.push(front->right);
            }
            vv.push_back(v);//将遍历完的一层数据存入
            levelsize=q.size();
        }
        return vv;
    }
};

三、二叉树的层序遍历 II

107. 二叉树的层序遍历 II - 力扣(LeetCode)

解题思路:该题是上题的变形题,很多同学容易在这题中想的过多,这一题想要的结果是上题的结果的逆置,所以按上题思路处理,最后将其结果逆置一下即可:

解题代码:

class Solution {
public:
    vector> levelOrderBottom(TreeNode* root) {
        queue q;
        int levelsize=0;
        if(root)//根节点入队列
        {
            q.push(root);
            levelsize=1;
        }
        vector> vv;
        while(!q.empty())
        {
            vector v;//记录每一层的数据
            while(levelsize--)//将上一层节点出队列的同时,将其孩子节点都入队列
            {
                TreeNode* front=q.front();
                q.pop();
                v.push_back(front->val);
                if(front->left)
                    q.push(front->left);
                if(front->right)
                    q.push(front->right);
            }
            vv.push_back(v);//将遍历完的一层数据存入
            levelsize=q.size();
        }
        reverse(vv.begin(),vv.end());//将结果逆置
        return vv;
    }
};

四、二叉树的最近公共祖先

236. 二叉树的最近公共祖先 - 力扣(LeetCode)

解题思路:对于该题我们需要了解两个节点最近公共祖先的特性,即这两个节点分别在最近公共祖先节点的左右子树;所以我们只要找到满足该条件的节点就找到了最近公共祖先节点

解题代码:

class Solution {
public:
    bool IsIntree(TreeNode* root, TreeNode* x)
    {
        if(root==nullptr)
            return false;
        if(root==x)
            return true;
        return IsIntree(root->left,x)||IsIntree(root->right,x);
    }
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if(root==p||root==q)
            return root;
        bool pInleft=IsIntree(root->left,p);
        bool pInright=!pInleft;
        bool qInleft=IsIntree(root->left,q);
        bool qInright=!qInleft;
        if((pInright&&qInleft)||(pInleft&&qInright))
        {
            return root;
        }
        else if(pInleft&&qInleft)
        {
            return lowestCommonAncestor(root->left,p,q);
        }
        else
        {
            return lowestCommonAncestor(root->right,p,q);
        }
    }
};

但是该方法最坏情况下的时间复杂度为O(N^2),运行时长并不理想:

【LeetCode】二叉树OJ_第2张图片

那下面我们换个思路来优化一下:我们可以记录从根节点到两个要查找节点的路径,从而将其转换为链表相交问题(对于链表的相交问题我们之前在LeetCode和牛客网经典链表题目合集 讲过),对于路径的存储我们可以用到栈来记录,最终找到共同的最近祖先节点:

class Solution {
public:
    bool GetPath(TreeNode* root, TreeNode* x,stack& s)
    {
        if(root==nullptr)
            return false;
        s.push(root);
        if(root==x)
            return true;
        if(GetPath(root->left,x,s)||GetPath(root->right,x,s))
            return true;
        s.pop();
        return false;
    }
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        stack pPath,qPath;
        GetPath(root,p,pPath);
        GetPath(root,q,qPath);
        while(pPath.size()!=qPath.size())
        {
            if(pPath.size()>qPath.size())
                pPath.pop();
            else
                qPath.pop();
        }
        while(pPath.top()!=qPath.top())
        {
            pPath.pop();
            qPath.pop();
        }
        return pPath.top();
    }
};

这样子时间复杂度就降到了O(N):

【LeetCode】二叉树OJ_第3张图片

五、二叉搜索树与双向链表

二叉搜索树与双向链表_牛客题霸_牛客网 (nowcoder.com)

解题思路:在该题中需要利用二叉搜索树中序遍历的结果是升序的,所以我们在中序遍历时修改所在节点与前驱节点的指针指向即可;用cur指针表示当前节点,用prev指针表示中序遍历时当前节点的上一个遍历节点,使cur指针的左孩子:

下面是一个二叉搜索树中序遍历时,修改当前节点和前驱节点的指针指向的演示:

【LeetCode】二叉树OJ_第4张图片【LeetCode】二叉树OJ_第5张图片【LeetCode】二叉树OJ_第6张图片【LeetCode】二叉树OJ_第7张图片【LeetCode】二叉树OJ_第8张图片【LeetCode】二叉树OJ_第9张图片

解题代码:

class Solution {
public:
	void InOrder(TreeNode* cur,TreeNode*& prev)
	{
		if(cur==nullptr)
			return;
		InOrder(cur->left,prev);
		cur->left=prev;
		if(prev)
			prev->right=cur;//让前驱节点指针的右孩子指向当前节点
		prev=cur;//更新前驱节点指针位置
		InOrder(cur->right,prev);
	}
    TreeNode* Convert(TreeNode* pRootOfTree) 
	{
		if(pRootOfTree==nullptr)
			return nullptr;
		TreeNode* prev=nullptr;
        InOrder(pRootOfTree,prev);//中序遍历的同时修改二叉树节点指针的指向
		TreeNode* head=pRootOfTree;//pRootOfTree并非链表的头节点,需要找到头节点后返回
		while(head->left)
		{
			head=head->left;
		}
		return head;
    }
};

六、从前序与中序遍历序列构造二叉树

105. 从前序与中序遍历序列构造二叉树 - 力扣(LeetCode)

解题思路:

对于任意一颗树而言,前序遍历的形式总是

[ 根节点, [左子树的前序遍历结果], [右子树的前序遍历结果] ]

即根节点总是前序遍历中的第一个节点。而中序遍历的形式总是

[ [左子树的中序遍历结果], 根节点, [右子树的中序遍历结果] ]

只要我们在中序遍历中定位到根节点,那么我们就可以分别知道左子树和右子树中的节点数目。由于同一颗子树的前序遍历和中序遍历的长度显然是相同的,因此我们就可以对应到前序遍历的结果中,对上述形式中的所有左右括号进行定位

这样以来,我们就知道了左子树的前序遍历和中序遍历结果,以及右子树的前序遍历和中序遍历结果,我们就可以递归地对构造出左子树和右子树,再将这两颗子树接到根节点的左右位置

解题代码:

class Solution {
public:

    TreeNode* _buildTree(vector& preorder, vector& inorder,int begini,int endi,int& n) 
    {
        if(begini>endi)
            return nullptr;
        TreeNode* newnode=new TreeNode(preorder[n]);//先利用先序遍历确定根节点
        int rooti=begini;
        while(rooti<=endi)//找到确定的根节点在中序遍历中的位置
        {
            if(inorder[rooti]==preorder[n])
                break;
            else
                ++rooti;
        }
        ++n;
        newnode->left=_buildTree(preorder,inorder,begini,rooti-1,n);//将剩下的左区间的节点连接到左子树上
        newnode->right=_buildTree(preorder,inorder,rooti+1,endi,n);//将剩下的右区间的节点连接到右子树上
        return newnode;
    }

    TreeNode* buildTree(vector& preorder, vector& inorder) {
        int n=0;
        return _buildTree(preorder,inorder,0,preorder.size()-1,n);
    }
};

七、从中序与后序遍历序列构造二叉树

106. 从中序与后序遍历序列构造二叉树 - 力扣(LeetCode)

解题思路:该题和上一题是同类型题,我们使用后序遍历的逆序来确定根节点,再在中序遍历中找到其对应的节点,再递归地对构造出右子树和左子树(后序遍历的逆序是相当于先遍历右子树的前序遍历,所以我们先要构建右子树再构建左子树)

解题代码:

class Solution {
public:
    TreeNode* _buildTree(vector& inorder,vector& postorder,int begini,int endi,int& n) 
    {
        if(begini>endi)
            return nullptr;
        TreeNode* newnode=new TreeNode(postorder[n]);//先利用后序遍历确定根节点
        int rooti=begini;
        while(rooti<=endi)//找到确定的根节点在中序遍历中的位置
        {
            if(inorder[rooti]==postorder[n])
                break;
            else
                ++rooti;
        }
        --n;
        newnode->right=_buildTree(inorder,postorder,rooti+1,endi,n);//将剩下的右区间的节点连接到右子树上
        newnode->left=_buildTree(inorder,postorder,begini,rooti-1,n);//将剩下的左区间的节点连接到左子树上
        return newnode;
    }
    TreeNode* buildTree(vector& inorder, vector& postorder) {
        int n=inorder.size()-1;
        return _buildTree(inorder,postorder,0,inorder.size()-1,n);
    }
};

你可能感兴趣的:(LeetCode,leetcode,算法,职场和发展)