LeetCode-Solutions/Python/remove-comments.py at master · pythonpeixun/LeetCode-Solutions · GitHub

30 lines (28 loc) · 999 Bytes

# 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

Expand file tree