Path Sum II 두 갈래 트리 경로 및 모든 경로 인쇄@LeetCode

3328 단어
제목:
두 갈래 나무와 하나의 값sum를 지정하여 이 두 갈래 나무가 루트에서 잎 노드까지의 경로가 경로와sum와 같은 것을 만족시킬 수 있는지 확인합니다.
모든 경로 인쇄
생각:
경전은 차례로 거슬러 올라간다!
package Level2;

import java.util.ArrayList;

import Utility.TreeNode;

/**
 * Path Sum II  
 * 
 * Given a binary tree and a sum, find all root-to-leaf paths where each path's sum equals the given sum.

For example:
Given the below binary tree and sum = 22,
              5
             / \
            4    8
           /     / \
          11  13  4
         /  \      / \
        7    2   5   1
return
[
   [5,4,11,2],
   [5,8,4,5]
]
 */


public class S113 {

	public static void main(String[] args) {
		TreeNode root = new TreeNode(5);
		TreeNode n1 = new TreeNode(4);
		TreeNode n2 = new TreeNode(8);
		root.left = n1;
		root.right = n2;
		TreeNode n3 = new TreeNode(11);
		TreeNode n4 = new TreeNode(13);
		TreeNode n5 = new TreeNode(4);
		n1.left = n3;
		n2.left = n4;
		n2.right = n5;
		TreeNode n6 = new TreeNode(7);
		TreeNode n7 = new TreeNode(2);
		n3.left = n6;
		n3.right = n7;
		TreeNode n8 = new TreeNode(5);
		TreeNode n9 = new TreeNode(1);
		n5.left = n8;
		n5.right = n9;
		
		ArrayList<ArrayList<Integer>> list = pathSum(root, 22);
		System.out.println(list);
	}
	
	public static ArrayList<ArrayList<Integer>> pathSum(TreeNode root, int sum) {
		ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();
		ArrayList<Integer> l =new ArrayList<Integer>();
		dfs(root, sum, list, l);
		return list;
    }
	
	//  
	private static void dfs(TreeNode root, int sum, ArrayList<ArrayList<Integer>> list, ArrayList<Integer> l){
		if(root == null){
			return;
		}
	
		//  
		if(root.val==sum && root.left==null && root.right==null){
			l.add(root.val);
			
			//  ! , !
			//  
			ArrayList<Integer> clone = new ArrayList<Integer>(l);
			list.add(clone);
			
			l.remove(l.size()-1);		//  // 
			return;
		}
	
		l.add(root.val);
		dfs(root.left, sum-root.val, list, l);
		dfs(root.right, sum-root.val, list, l);
		l.remove(l.size()-1);		//  // 
	}
}

귀속된basecase는 루트가null인 상황뿐만 아니라 잎 노드의 상황도 고려해야 합니다!
/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public List<List<Integer>> pathSum(TreeNode root, int sum) {
        List<List<Integer>> ret = new ArrayList<List<Integer>>();
        List<Integer> list = new ArrayList<Integer>();
        rec(ret, list, root, sum);
        return ret;
    }
    
    public void rec(List<List<Integer>> ret, List<Integer> list, TreeNode root, int sum) {
        if(root == null) {
            return;
        }
        if(root.left==null && root.right==null && sum == root.val){
            list.add(root.val);
            ret.add(new ArrayList<Integer>(list));
            list.remove(list.size()-1);
            return;
        }
        
        list.add(root.val);
        rec(ret, list, root.left, sum-root.val);
        rec(ret, list, root.right, sum-root.val);
        list.remove(list.size()-1);
    }
}

좋은 웹페이지 즐겨찾기