自己写遇到瓶颈,子节点交换完无法返回到pRoot当中
import java.util.*;
/*
* public class TreeNode {
* int val = 0;
* TreeNode left = null;
* TreeNode right = null;
* public TreeNode(int val) {
* this.val = val;
* }
* }
*/
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param pRoot TreeNode类
* @return TreeNode类
*/
public TreeNode Mirror (TreeNode pRoot) {
// write code here
TreeNode left=pRoot.left;
TreeNode right=pRoot.right;
//交换左边
while(left!=null||right!=null)
{
int cen=pRoot.left.val;
pRoot.left.val=pRoot.right.val;
pRoot.right.val=cen;
//如果左边指针不为空,并且左右有儿子,则调用Mirror
if(left!=null)
{
if(left.left!=null||left.right!=null)
{
Mirror(left);
}
}
if(right!=null)
{ if(right.left!=null||right.right!=null)
{
Mirror(right);
}
}
}
//返回中间节点
return pRoot;
}
}
修改方案
递归时最好不使用while
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param pRoot TreeNode类
* @return TreeNode类
*/
public TreeNode Mirror (TreeNode pRoot) {
// write code here
if(pRoot==null)
{
return null;
}
TreeNode left=pRoot.left;
TreeNode right=pRoot.right;
if(left==null&&right==null)
{
return pRoot;
}
TreeNode cen=pRoot.left;
pRoot.left=pRoot.right;
pRoot.right=cen;
//如果左边指针不为空,并且左右有儿子,则调用Mirror
if(left!=null)
{
Mirror(left);
}
if(right!=null)
{
Mirror(right);
}
//返回中间节点
return pRoot;
}
}