【力扣刷题】重新格式化电话号码

1694.重新格式化电话号码
给你一个字符串形式的电话号码 number 。number 由数字、空格 ’ '、和破折号 ‘-’ 组成。
请你按下述方式重新格式化电话号码。
首先，删除 所有的空格和破折号。
其次，将数组从左到右 每 3 个一组 分块，直到 剩下 4 个或更少数字。剩下的数字将按下述规定再分块：
2 个数字：单个含 2 个数字的块。
3 个数字：单个含 3 个数字的块。
4 个数字：两个分别含 2 个数字的块。
最后用破折号将这些块连接起来。注意，重新格式化过程中 不应该 生成仅含 1 个数字的块，并且 最多 生成两个含 2 个数字的块。
返回格式化后的电话号码。

我的代码：

class Solution(object):
    def reformatNumber(self, number):
        """
        :type number: str
        :rtype: str
        """
        if not number:
            return ""

        mark = 0  # 记录倒数四个数的位置
        num = ""  # 纯数字矩阵
        ret = ""  # 返回的结果矩阵
        count = 0  # 记录数字段

        for n in number:
            if n.isdigit():
                num += n  # 将纯数字加入

        if len(num) % 3 == 0:
            for i in range(len(num)):
                ret += num[i]
                count += 1
                if count % 3 == 0 and i != len(num) - 1:
                    ret += "-"  # 隔三个数字加-
                    count = 0
        elif len(num) < 3:
            ret = num
        else:
            for i in range(len(num)):
                if len(num) - i == 4:  # 倒数四个数
                    mark = i
                    count = 0
                    break
                ret += num[i]
                count += 1
                if count % 3 == 0:
                    ret += "-"  # 隔三个数字加-
                    count = 0
            for i in range(mark, len(num)):
                ret += num[i]
                count += 1
                if count % 2 == 0 and i != len(num) - 1:
                    ret += "-"
                    count = 0
        return ret                 
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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

本来想着能够尽可能的减少空间消耗，但是无奈能力有限，只能通过笨办法for把答案输出了。

需要注意的地方：
①最后一个位置不能加‘-’；
②注意字符串数字较短的情况；

结果：