跳转至

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]