【模拟】重新格式化电话号码

题目

1694. 重新格式化电话号码

---

给你一个字符串形式的电话号码 number 。number 由数字、空格 ' '、和破折号 '-' 组成。

请你按下述方式重新格式化电话号码。

• 首先，删除 所有的空格和破折号。
• 其次，将数组从左到右 每 3 个一组 分块，直到 剩下 4 个或更少数字。剩下的数字将按下述规定再分块：
  • 2 个数字：单个含 2 个数字的块。
  • 3 个数字：单个含 3 个数字的块。
  • 4 个数字：两个分别含 2 个数字的块。

最后用破折号将这些块连接起来。注意，重新格式化过程中 不应该 生成仅含 1 个数字的块，并且 最多 生成两个含 2 个数字的块。

返回格式化后的电话号码。

示例 1：

输入：number = "1-23-45 6"
输出："123-456"
解释：数字是 "123456"
步骤 1：共有超过 4 个数字，所以先取 3 个数字分为一组。第 1 个块是 "123" 。
步骤 2：剩下 3 个数字，将它们放入单个含 3 个数字的块。第 2 个块是 "456" 。
连接这些块后得到 "123-456" 。

示例 2：

输入：number = "123 4-567"
输出："123-45-67"
解释：数字是 "1234567".
步骤 1：共有超过 4 个数字，所以先取 3 个数字分为一组。第 1 个块是 "123" 。
步骤 2：剩下 4 个数字，所以将它们分成两个含 2 个数字的块。这 2 块分别是 "45" 和 "67" 。
连接这些块后得到 "123-45-67" 。

示例 3：

输入：number = "123 4-5678"
输出："123-456-78"
解释：数字是 "12345678" 。
步骤 1：第 1 个块 "123" 。
步骤 2：第 2 个块 "456" 。
步骤 3：剩下 2 个数字，将它们放入单个含 2 个数字的块。第 3 个块是 "78" 。
连接这些块后得到 "123-456-78" 。

示例 4：

输入：number = "12"
输出："12"

示例 5：

输入：number = "--17-5 229 35-39475 "
输出："175-229-353-94-75"

提示：

• 2 <= number.length <= 100
• number 由数字和字符 '-' 及 ' ' 组成。
• number 中至少含 2 个数字。

解题

方法一：模拟

思路

维护一个计数(cnt)，遍历字符串遇到数字就把它拼到答案字符串中(ans)，如果计数是 $3$ 的倍数就在 ans 中拼入一个 '-'，遍历完成后，如果 cnt 能被 $3$ 除尽，那答案字符串就被拼成了形如 xxx-xxx- 的形式，需要把最后一个 '-' 删掉；而如果 cnt 除以 $3$ 余一，就说明答案字符串被拼成了形如 xxx-xxx-x 的形式，需要改成 xxx-xx-xx ，其他情况 ans 均符合要求。

代码

class Solution {
    public String reformatNumber(String number) {
        StringBuilder ans = new StringBuilder();
        int cnt = 0;
        for (char ch : number.toCharArray()) {
            if (ch == ' ' || ch == '-') continue;
            ++cnt;
            ans.append(ch);
            if (cnt % 3 == 0) ans.append('-');
        }
        if (cnt % 3 == 0) ans.deleteCharAt(ans.length() - 1);
        else if (cnt % 3 == 1) ans.deleteCharAt(ans.length() - 2).insert(ans.length() - 2, '-');
        return ans.toString();
    }
}