Question

1694. 重新格式化电话号码

English Version

题目描述

给你一个字符串形式的电话号码 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 个数字。

解法

方法一：简单模拟

我们先按照题意，去除字符串中的所有空格和破折号。

记当前字符串长度为 $n$，然后从前往后遍历字符串，每 $3$ 个字符为一组，将其加入结果字符串中，共取 $n / 3$ 组。

若最后剩余 $1$ 个字符，则将前面的最后一组的最后一个字符与该字符组成一个新的两个字符的组，加入结果字符串中；若最后剩余 $2$ 个字符，直接将这连个字符组成一个新的组，加入结果字符串中。

最后将所有组之间加上破折号，返回结果字符串即可。

时间复杂度 $O(n)$，空间复杂度 $O(n)$。其中 $n$ 为字符串长度。

class Solution:
  def reformatNumber(self, number: str) -> str:
    number = number.replace("-", "").replace(" ", "")
    n = len(number)
    ans = [number[i * 3 : i * 3 + 3] for i in range(n // 3)]
    if n % 3 == 1:
      ans[-1] = ans[-1][:2]
      ans.append(number[-2:])
    elif n % 3 == 2:
      ans.append(number[-2:])
    return "-".join(ans)

class Solution {
  public String reformatNumber(String number) {
    number = number.replace("-", "").replace(" ", "");
    int n = number.length();
    List<String> ans = new ArrayList<>();
    for (int i = 0; i < n / 3; ++i) {
      ans.add(number.substring(i * 3, i * 3 + 3));
    }
    if (n % 3 == 1) {
      ans.set(ans.size() - 1, ans.get(ans.size() - 1).substring(0, 2));
      ans.add(number.substring(n - 2));
    } else if (n % 3 == 2) {
      ans.add(number.substring(n - 2));
    }
    return String.join("-", ans);
  }
}

class Solution {
public:
  string reformatNumber(string number) {
    string s;
    for (char c : number) {
      if (c != ' ' && c != '-') {
        s.push_back(c);
      }
    }
    int n = s.size();
    vector<string> res;
    for (int i = 0; i < n / 3; ++i) {
      res.push_back(s.substr(i * 3, 3));
    }
    if (n % 3 == 1) {
      res.back() = res.back().substr(0, 2);
      res.push_back(s.substr(n - 2));
    } else if (n % 3 == 2) {
      res.push_back(s.substr(n - 2));
    }
    string ans;
    for (auto& v : res) {
      ans += v;
      ans += "-";
    }
    ans.pop_back();
    return ans;
  }
};

func reformatNumber(number string) string {
  number = strings.ReplaceAll(number, " ", "")
  number = strings.ReplaceAll(number, "-", "")
  n := len(number)
  ans := []string{}
  for i := 0; i < n/3; i++ {
    ans = append(ans, number[i*3:i*3+3])
  }
  if n%3 == 1 {
    ans[len(ans)-1] = ans[len(ans)-1][:2]
    ans = append(ans, number[n-2:])
  } else if n%3 == 2 {
    ans = append(ans, number[n-2:])
  }
  return strings.Join(ans, "-")
}

function reformatNumber(number: string): string {
  const cs = [...number].filter(c => c !== ' ' && c !== '-');
  const n = cs.length;
  return cs
    .map((v, i) => {
      if (((i + 1) % 3 === 0 && i < n - 2) || (n % 3 === 1 && n - 3 === i)) {
        return v + '-';
      }
      return v;
    })
    .join('');
}

impl Solution {
  pub fn reformat_number(number: String) -> String {
    let cs: Vec<char> = number
      .chars()
      .filter(|&c| c != ' ' && c != '-')
      .collect();
    let n = cs.len();
    cs.iter()
      .enumerate()
      .map(|(i, c)| {
        if ((i + 1) % 3 == 0 && i < n - 2) || (n % 3 == 1 && i == n - 3) {
          return c.to_string() + &"-";
        }
        c.to_string()
      })
      .collect()
  }
}