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

leetcode 226: Invert Binary Tree

时间:2015-09-20 19:04:39      阅读:158      评论:0      收藏:0      [点我收藏+]

标签:

题目:

Invert a binary tree.

     4
   /     2     7
 / \   / 1   3 6   9

to

     4
   /     7     2
 / \   / 9   6 3   1

Trivia:
This problem was inspired by this original tweet by Max Howell:

Google: 90% of our engineers use the software you wrote (Homebrew), but you can’t invert a binary tree on a whiteboard so fuck off.

 

思路:

置换,用递归,左右子树置换

代码:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public TreeNode invertTree(TreeNode root) {
        if(root == null) 
            return root;
        else {
            if(root.left!=null && root.right!=null) {
                TreeNode temp;
                temp = root.left;
                root.left = root.right;
                root.right = temp;
                
                invertTree(root.left);
                invertTree(root.right);
            } else if(root.left == null && root.right !=null) {
                root.left = root.right;
                root.right = null;
                invertTree(root.left);
            } else if(root.left != null && root.right ==null) {
                root.right = root.left;
                root.left = null;
                invertTree(root.right);
            } else {
                return root;
            }
        }
        return root;
    }
}

 

leetcode 226: Invert Binary Tree

标签:

原文地址:http://www.cnblogs.com/cactus1504/p/4823790.html

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