forked from kamyu104/LeetCode-Solutions
-
Notifications
You must be signed in to change notification settings - Fork 0
/
remove-comments.py
30 lines (28 loc) · 999 Bytes
/
remove-comments.py
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
# Time: O(n), n is the length of the source
# Space: O(k), k is the max length of a line
class Solution(object):
def removeComments(self, source):
"""
:type source: List[str]
:rtype: List[str]
"""
in_block = False
result, newline = [], []
for line in source:
i = 0
while i < len(line):
if not in_block and i+1 < len(line) and line[i:i+2] == '/*':
in_block = True
i += 1
elif in_block and i+1 < len(line) and line[i:i+2] == '*/':
in_block = False
i += 1
elif not in_block and i+1 < len(line) and line[i:i+2] == '//':
break
elif not in_block:
newline.append(line[i])
i += 1
if newline and not in_block:
result.append("".join(newline))
newline = []
return result