剑指编程（12）

一、
给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。
注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。

class TreeLinkNode {
    int val;
    TreeLinkNode left = null;
    TreeLinkNode right = null;
    TreeLinkNode next = null;

    TreeLinkNode(int val) {
        this.val = val;
    }
}
public class Solution {
    public TreeLinkNode GetNext(TreeLinkNode pNode) {
        //如果有右子树，下一个结点肯定在右子树中
        if(pNode.right != null) {
            TreeLinkNode tmp = pNode.right;
            while(tmp.left != null) {
                tmp = tmp.left;
            }
            return tmp;
        }
        //往前遍历直接找到直接父节点
        while(pNode.next != null) {
            if(pNode.next.left == pNode) {
                return pNode.next;
            }
            pNode = pNode.next;
        }
        return null;
    }
}

二、
请实现一个函数，用来判断一颗二叉树是不是对称的。
注意，如果一个二叉树同此二叉树的镜像是同样的，定义其为对称的。

class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;
    }
}
public class Solution {
    //左节点的值和右节点的值相等即可
    //递归到左子树和右子树
    boolean isSymmetrical(TreeNode pRoot) {
        if(pRoot == null) {
            return true;
        }
        return isSymmetrical(pRoot.left, pRoot.right);
    }
    //涉及对称，还是应该重载
    boolean isSymmetrical(TreeNode node1, TreeNode node2) {
        if(node1 == null && node2 == null) {
            return true;
        }
        if(node1 != null && node2 != null) {
            if(node1.val == node2.val) {
                return isSymmetrical(node1.left, node2.right) && isSymmetrical(node1.right, node2.left);
            }
        }
        return false;
    }
}

三、
从上到下按层打印二叉树，同一层结点从左至右输出。
每一层输出一行。

class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;
    }
}
public class Solution {
    //层序遍历，用队列实现
    ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        if(pRoot == null) {
            return res;
        }
        Queue<TreeNode> q = new LinkedList<>();
        q.add(pRoot);
        while(!q.isEmpty()) {
            //用于记录一层
            int index = 0, size = q.size();
            ArrayList<Integer> inner = new ArrayList<>();
            while(index < size) {
                TreeNode cur = q.poll();
                inner.add(cur.val);
                if(cur.left != null) {
                    q.add(cur.left);
                }
                if(cur.right != null) {
                    q.add(cur.right);
                }
                index++;
            }
            res.add(inner);
        }
        return res;
    }
}

四、
请实现一个函数按照之字形打印二叉树，
即第一行按照从左到右的顺序打印，第二层按照从右至左的顺序打印，
第三行按照从左到右的顺序打印，其他行以此类推。

class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;
    }
}
public class Solution {
    //层序遍历，一开始从左到右，之后和从右到左交替
    public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        if(pRoot == null) {
            return res;
        }
        Queue<TreeNode> q = new LinkedList<>();
        q.add(pRoot);
        boolean flag = true;
        while(!q.isEmpty()) {
            //用于记录一层
            int index = 0, size = q.size();
            ArrayList<Integer> inner = new ArrayList<>();
            while(index < size) {
                TreeNode cur = q.poll();
                inner.add(cur.val);
                if(cur.left != null) {
                    q.add(cur.left);
                }
                if(cur.right != null) {
                    q.add(cur.right);
                }
                index++;
            }
            if(flag) {
                res.add(inner);
                flag = false;
            }else {
                //将元素反过来
                ArrayList<Integer> rev = new ArrayList<>();
                int len = inner.size();
                for(int i = len - 1; i >= 0; i--) {
                    rev.add(inner.get(i));
                }
                res.add(rev);
                flag = true;
            }
        }
        return res;
    }
}

五、
给定一颗二叉搜索树，请找出其中的第k大的结点。
例如:
5
/ \
3 7
/\ /\
2 4 6 8 中，按结点数值大小顺序第三个结点的值为4。

class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;
    }
}
public class Solution {
    //二叉搜索树中序遍历是从小到大排序的
    int cnt = 0;
    TreeNode KthNode(TreeNode pRoot, int k) {
        if(pRoot != null) {
            TreeNode tmp;
            if((tmp = KthNode(pRoot.left, k)) != null) {
                return tmp;
            }
            cnt++;
            if(k == cnt) {
                return pRoot;
            }
            if((tmp = KthNode(pRoot.right, k)) != null) {
                return tmp;
            }
        }
        return null;
    }
}