java算法(九)

1、從上往下打印出二叉樹的每個節點,同層節點從左至右打印。

import java.util.*;
public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        ArrayList<Integer> list = new ArrayList<>();
        if(root == null) return list;
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        list.add(root.val);
        
        while(!queue.isEmpty()){
            TreeNode node = queue.poll();
            if(node.left != null) {
                list.add(node.left.val);
                queue.add(node.left);
            }
            if(node.right != null) {
                list.add(node.right.val);
                queue.add(node.right);
            }
        }
        
        return list;
        
    }
}

2、輸入一個鏈表,按鏈表從尾到頭的順序返回一個ArrayList。

import java.util.Stack;
import java.util.ArrayList;
public class Solution {
    public ArrayList<Integer> printListFromTailToHead(ListNode listNode) {
        Stack<Integer> stack = new Stack<>();
        while(listNode != null){
            stack.push(listNode.val);
            listNode = listNode.next;
        }
        
        ArrayList<Integer> list = new ArrayList<>();
        while(!stack.isEmpty()){
            list.add(stack.pop());
        }
        return list;
    }
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章