forked from onlybooks/java-algorithm-interview
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathP90_2.java
More file actions
45 lines (40 loc) · 1.87 KB
/
P90_2.java
File metadata and controls
45 lines (40 loc) · 1.87 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
package ch22;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
public class P90_2 {
HashMap<String, List<Integer>> memo = new HashMap<>();
public List<Integer> diffWaysToCompute(String expression) {
List<Integer> result = new ArrayList<>();
// 이미 계산해둔 동일한 표현식이라면 기존 결과 재활용
if (memo.containsKey(expression))
return memo.get(expression);
for (int i = 0; i < expression.length(); i++) {
char c = expression.charAt(i);
// 연산자가 등장하면 좌, 우 분할하여 각각 계산 결과를 리턴받아 처리한다.
if (c == '+' || c == '-' || c == '*') {
// 연산자가 등장한 위치를 기준으로 좌, 우 분할
List<Integer> left = diffWaysToCompute(expression.substring(0, i));
List<Integer> right = diffWaysToCompute(expression.substring(i + 1));
// 좌, 우 분할된 표현식의 연산 결과 리스트를 받아서 모두 순회하며 연산 처리
for (int l : left) {
for (int r : right) {
if (c == '+') {
result.add(l + r);
} else if (c == '-') {
result.add(l - r);
} else if (c == '*') {
result.add(l * r);
}
}
}
}
}
// 연산 결과가 없다는 것은 연산자가 포함되어 있지 않다는 뜻이며, 정숫값만 있는 경우이므로 이 값만 담아서 리턴
if (result.isEmpty())
result.add(Integer.parseInt(expression));
// 계산 결과는 추후 재활용을 위해 저장
memo.put(expression, result);
return result;
}
}