199. 두 갈래 나무의 오른쪽 보기

1273 단어 LeetCode

두 갈래 나무의 오른쪽 보기


두 갈래 나무를 정해서 오른쪽에 서 있는 것을 상상하고 꼭대기에서 끝까지 순서대로 오른쪽에서 볼 수 있는 노드 값을 되돌려줍니다.
예:
 : [1,2,3,null,5,null,4]
 : [1, 3, 4]
 :

   1            

사고방식 + 코드 + 주석:
public List rightSideView(TreeNode root) {
        /*
         : Queue, , Queue , , count, , 
         , 
         */
        List res=new ArrayList<>();
        if (root==null)
        {
            return res;
        }
        Queue queue=new LinkedList<>();
        queue.add(root);
        while (queue.size()>0)
        {
            int count=0;
            int size=queue.size();
            while (queue.size()>0)
            {
                TreeNode node=((LinkedList) queue).pop();
                if (node.left!=null)
                {
                    queue.add(node.left);
                }
                if (node.right!=null)
                {
                    queue.add(node.right);
                }
                count++;
                if (count==size)
                {
                    res.add(node.val);
                    break;
                }
            }
        }
        return res;
    }

좋은 웹페이지 즐겨찾기