給定一棵二叉樹,想象自己站在它的右側,按照從頂部到底部的順序,返回從右側所能看到的節點值。
示例:
輸入: [1,2,3,null,5,null,4]
輸出: [1, 3, 4]
解釋:1 <---
/ \
2 3 <---
\ \
5 4 <---
想法與思路:
1.因爲右視圖是同一層最右邊的節點
1.構建Map Key 層高 , Value 節點值
2.想象中序遍歷,右邊的節點最後遍歷到,map.put 相同key 覆蓋,所以留在最後的一定就是同層最右邊節點
3.樹不會太深,不知道深度,嘗試從0層遍歷到最大層 ,取出 map的 value
代碼
package com.offer.test.leetcode;
import java.util.*;
/**
* Created by szh on 2020/6/8.
*/
public class Leetcode_199_TreeRightView {
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) {
val = x;
}
}
public List<Integer> rightSideView(TreeNode root) {
List<Integer> list = new ArrayList<>();
Map<Integer,Integer> map = new HashMap<>();
map.values();
if(root != null){
traverse(root, 0, map);
}
int i = 0;
boolean flag = true;
while(flag){
if(map.containsKey(i)){
list.add(map.get(i));
++i;
}else{
flag = false;
}
}
return list;
}
public void traverse(TreeNode curRoot, int deep, Map<Integer,Integer> map){
map.put(deep,curRoot.val);
if(curRoot.left != null){
traverse(curRoot.left, deep + 1, map);
}
if(curRoot.right != null){
traverse(curRoot.right, deep + 1, map);
}
}
}