跳转至

LeetCode: 722. 删除注释

1、题目描述

给一个 C++ 程序，删除程序中的注释。这个程序source是一个数组，其中source[i]表示第i行源码。 这表示每行源码由\n分隔。

在 C++ 中有两种注释风格，行内注释和块注释。

字符串// 表示行注释，表示//和其右侧的其余字符应该被忽略。

字符串/* 表示一个块注释，它表示直到*/的下一个（非重叠）出现的所有字符都应该被忽略。（阅读顺序为从左到右）非重叠是指，字符串/*/并没有结束块注释，因为注释的结尾与开头相重叠。

第一个有效注释优先于其他注释：如果字符串//出现在块注释中会被忽略。 同样，如果字符串/*出现在行或块注释中也会被忽略。

如果一行在删除注释之后变为空字符串，那么不要输出该行。即，答案列表中的每个字符串都是非空的。

样例中没有控制字符，单引号或双引号字符。比如，source = "string s = "/* Not a comment. */";" 不会出现在测试样例里。（此外，没有其他内容（如定义或宏）会干扰注释。）

我们保证每一个块注释最终都会被闭合， 所以在行或块注释之外的/*总是开始新的注释。

最后，隐式换行符可以通过块注释删除。 有关详细信息，请参阅下面的示例。

从源代码中删除注释后，需要以相同的格式返回源代码。

示例 1:

输入: 
source = ["/*Test program */", "int main()", "{ ", "  // variable declaration ", "int a, b, c;", "/* This is a test", "   multiline  ", "   comment for ", "   testing */", "a = b + c;", "}"]

示例代码可以编排成这样:
/*Test program */
int main()
{ 
  // variable declaration 
int a, b, c;
/* This is a test
   multiline  
   comment for 
   testing */
a = b + c;
}

输出: ["int main()","{ ","  ","int a, b, c;","a = b + c;","}"]

编排后:
int main()
{ 

int a, b, c;
a = b + c;
}

解释: 
第 1 行和第 6-9 行的字符串 /* 表示块注释。第 4 行的字符串 // 表示行注释。

示例 2:

输入: 
source = ["a/*comment", "line", "more_comment*/b"]
输出: ["ab"]
解释: 原始的 source 字符串是 "a/*comment\nline\nmore_comment*/b", 其中我们用粗体显示了换行符。删除注释后，隐含的换行符被删除，留下字符串 "ab" 用换行符分隔成数组时就是 ["ab"].

注意:

  • source的长度范围为[1, 100].
  • source[i]的长度范围为[0, 80].
  • 每个块注释都会被闭合。
  • 给定的源码中不会有单引号、双引号或其他控制字符。

2、解题思路

  • 对于单行注释和多行注释分别处理
  • 设置一个标志位，确认当前是否找到了多行注释的结束标记
class Solution:
    def removeComments(self, source: List[str]) -> List[str]:
        ans = []
        length = len(source)

        line = "//"
        many_left = "/*"
        many_right = "*/"

        pre = 0
        pos = 0
        temp = ""
        while pos < length:
            cur = source[pos]
            if pre:
                index = cur.find(many_right)
                if index != -1:
                    source[pos] = temp + cur[index + 2:]
                    pos -= 1
                    pre = 0
                else:
                    source[pos] = ""
            else:
                line_index = cur.find(line)
                many_index = cur.find(many_left)
                if line_index != -1 and many_index == -1 or (line_index != -1 and line_index < many_index):
                    source[pos] = cur[:line_index]
                elif line_index == -1 and many_index != -1 or (many_index != -1 and many_index < line_index):
                    temp_pos = cur.find(many_right, many_index + 2)
                    if temp_pos != -1:
                        source[pos] = cur[:many_index] + cur[temp_pos + 2:]
                        pos -= 1
                    else:
                        pre = 1
                        temp = cur[:many_index]
                        source[pos] = ""
            pos += 1

        return [x for x in source if x]