Leetcode Java Sum Root to Leaf Numbers
업데이트:
문제
코드
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public int sumNumbers(TreeNode root) {
return this.recursive(root, 0);
}
private int recursive(TreeNode node, int sum) {
if (node == null) {
return 0;
} else if (node.left == null && node.right == null) {
return sum * 10 + node.val;
} else {
return this.recursive(node.left, sum * 10 + node.val) + this.recursive(node.right, sum * 10 + node.val);
}
}
}
결과
설명
- 주어진 TreeNode인 root를 이용하여 마지막 node까지 이동하며 val 값을 순차적으로 이어준 결과들을 합친 값을 구하는 문제이다.
- 아래의 예제를 보면, 1 -> 2, 1 -> 3으로 되어, 12 + 13인 25가 문제의 결과가 된다.
1 2 3
- 아래의 예제를 보면, 1 -> 2, 1 -> 3으로 되어, 12 + 13인 25가 문제의 결과가 된다.
- 재귀 호출을 이용하여 root 노드에서 부터 최종 합을 구하여 문제의 결과로 반환한다.
- node가 null인 경우, 마지막 노드이므로 해당 val 값은 0으로 반환한다.
- node의 left와 right가 null인 경우, 기존 더해진 sum에 10을 곱하고, node의 val 값을 더하여 반환한다.
- 그 외의 경우 node의 left와 right를 재귀 호출한 값의 합을 반환한다.
- 최종 재귀 호출을 수행하여 순차적으로 이어준 값의 합을 주어진 문제의 결과로 반환한다.
소스
Sample Code는 여기에서 확인 가능합니다.
댓글남기기