blob: c4d2b3e22ec650d470a403ca720dd6651919c0bb (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
#!/usr/bin/env python
import typing
class Solution:
def minCost(self, colors: str, neededTime: typing.List[int]) -> int:
time = 0
n = len(colors)
for i in range(1, n):
if colors[i] == colors[i - 1]:
time += min(neededTime[i], neededTime[i - 1])
neededTime[i] = max(neededTime[i], neededTime[i - 1])
return time
def main():
solution = Solution()
print(solution.minCost("abaa", [1, 2, 3, 4]))
if __name__ == "__main__":
main()
|