지그재그로 두 갈래 나무를 인쇄하다

3605 단어 검지offer
제목 설명은 함수를 지그재그로 인쇄하는 두 갈래 트리, 즉 첫 번째 줄은 왼쪽에서 오른쪽으로, 두 번째 줄은 오른쪽에서 왼쪽으로, 세 번째 줄은 왼쪽에서 오른쪽으로, 다른 줄은 이와 같이 인쇄합니다.사고방식: 두 개의 창고, 한 개의 창고는 한 층의 노드를 저장하고 창고가 선진적으로 나온 특징에 따라 각 층이 인쇄한 노드와 넣은 노드의 순서를 반대로 할 수 있다.이것으로 각 층의'역순'의 특징을 달성한다.
public ArrayList> Print(TreeNode pRoot) {
        
        ArrayList> lists = new ArrayList>();
        if(pRoot == null){
            return lists;
        }
        
        //Queue queue1 = new LinkedList();
        Stack stack1 =  new Stack();
         Stack stack2 =  new Stack();

        stack1.push(pRoot);
        while(!stack1.isEmpty()||!stack2.isEmpty()){
             ArrayList list1 = new ArrayList();
            while(!stack1.isEmpty()){
                 TreeNode node = stack1.pop();
                if(node.left!=null){
                    stack2.push(node.left);
                }
                if(node.right!=null){
                    stack2.push(node.right);
                }
                list1.add(node.val);
            }
            if(list1.size()!=0){
                  lists.add(list1);
            }
           
            ArrayList list2 = new ArrayList();
            while(!stack2.isEmpty()){
               TreeNode node =  stack2.pop();
                if(node.right!=null){
                    stack1.push(node.right);
                }
                if(node.left!=null){
                    stack1.push(node.left);
                }
                list2.add(node.val);
            }
            if(list2.size()!=0){
                 lists.add(list2);
            }
        }
        return lists;
    }

또 하나의 변형 문제가 있다. 각 층 노드는 왼쪽에서 오른쪽으로 노드 제목을 인쇄하고 위에서 아래로 층별로 두 갈래 나무를 인쇄하며 같은 층 결점은 왼쪽에서 오른쪽으로 출력한다.층마다 줄을 출력합니다.사고방식: 대열의 선진적인 선출 사상을 이용하다
ArrayList > Print(TreeNode pRoot) {
        
   
        ArrayList > lists = new ArrayList >();
          if(pRoot==null){
            return lists;
            }
        Queue queue1 = new LinkedList();
        Queue queue2 = new LinkedList();
        queue1.offer(pRoot);
        
        while(!queue1.isEmpty()||!queue2.isEmpty()){
            ArrayList list1 = new ArrayList();
            while(!queue1.isEmpty()){
              TreeNode node =   queue1.poll();
                list1.add(node.val);
                if(node.left!=null){
                    queue2.offer(node.left);
                }
                if(node.right!=null){
                    queue2.offer(node.right);
                }
            }
            if(list1.size()!=0){
                lists.add(list1);
            }
            ArrayList list2 = new ArrayList();
            while(!queue2.isEmpty()){
               TreeNode node =  queue2.poll();
                list2.add(node.val);
                 if(node.left!=null){
                    queue1.offer(node.left);
                }
                if(node.right!=null){
                    queue1.offer(node.right);
                }
            }
            if(list2.size()!=0){
                  lists.add(list2);
            }
        }
        return lists;
    }

좋은 웹페이지 즐겨찾기