-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathsol.py
More file actions
63 lines (48 loc) · 1.4 KB
/
Copy pathsol.py
File metadata and controls
63 lines (48 loc) · 1.4 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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
INFINITY = 1000000000000
DP = None
S = None
def solve(start, end):
if DP[start][end] < INFINITY:
return DP[start][end]
DP[start][start] = 3
LAST_SEEN = {
S[start]: [start],
}
for k, letter in enumerate(S):
if k < start + 1:
continue
if k > end:
continue
if letter in LAST_SEEN:
if LAST_SEEN[letter][-1] == k - 1:
DP[start][k] = DP[start][k - 1] + 1
LAST_SEEN[letter].append(k)
continue
opt_a = DP[start][k - 1] + 3
opt_b = INFINITY
if letter in LAST_SEEN:
opt_b = INFINITY
for pos in LAST_SEEN[letter]:
alternative = DP[start][pos]
alternative += (
solve(
pos + 1,
k,
)
- 3
+ 1
)
opt_b = min(opt_b, alternative)
DP[start][k] = min(opt_a, opt_b)
if letter in LAST_SEEN:
LAST_SEEN[letter].append(k)
else:
LAST_SEEN[letter] = [k]
return DP[start][end]
if __name__ == "__main__":
n = int(input())
for _ in range(n):
S = input()
DP = [[INFINITY for _ in range(len(S) + 1)] for __ in range(len(S) + 1)]
solve(0, len(S))
print(DP[0][len(S) - 1])