Question

1946. 子字符串突变后可能得到的最大整数

English Version

题目描述

给你一个字符串 num ，该字符串表示一个大整数。另给你一个长度为 10 且 下标从 0  开始 的整数数组 change ，该数组将 0-9 中的每个数字映射到另一个数字。更规范的说法是，数字 d 映射为数字 change[d] 。

你可以选择 突变  num 的任一子字符串。突变 子字符串意味着将每位数字 num[i] 替换为该数字在 change 中的映射（也就是说，将 num[i] 替换为 change[num[i]]）。

请你找出在对 num 的任一子字符串执行突变操作（也可以不执行）后，可能得到的 最大整数 ，并用字符串表示返回。

子字符串 是字符串中的一个连续序列。

 

示例 1：

输入：num = "_1_32", change = [9,8,5,0,3,6,4,2,6,8]
输出："_8_32"
解释：替换子字符串 "1"：
- 1 映射为 change[1] = 8 。
因此 "_1_32" 变为 "_8_32" 。
"832" 是可以构造的最大整数，所以返回它的字符串表示。

示例 2：

输入：num = "_021_", change = [9,4,3,5,7,2,1,9,0,6]
输出："_934_"
解释：替换子字符串 "021"：
- 0 映射为 change[0] = 9 。
- 2 映射为 change[2] = 3 。
- 1 映射为 change[1] = 4 。
因此，"_021_" 变为 "_934_" 。
"934" 是可以构造的最大整数，所以返回它的字符串表示。 

示例 3：

输入：num = "5", change = [1,4,7,5,3,2,5,6,9,4]
输出："5"
解释："5" 已经是可以构造的最大整数，所以返回它的字符串表示。

 

提示：

• 1 <= num.length <= 105
• num 仅由数字 0-9 组成
• change.length == 10
• 0 <= change[d] <= 9

解法

方法一：贪心

从左到右遍历字符串 num，找到第一个比 change 中对应数字小的数字，然后将其替换为 change 中对应的数字，直到遇到比 change 中对应数字大的数字，停止替换。

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

class Solution:
  def maximumNumber(self, num: str, change: List[int]) -> str:
    s = list(num)
    for i, c in enumerate(s):
      if change[int(c)] > int(c):
        while i < len(s) and int(s[i]) <= change[int(s[i])]:
          s[i] = str(change[int(s[i])])
          i += 1
        break
    return ''.join(s)

class Solution {
  public String maximumNumber(String num, int[] change) {
    char[] s = num.toCharArray();
    for (int i = 0; i < s.length; ++i) {
      if (change[s[i] - '0'] > s[i] - '0') {
        for (; i < s.length && s[i] - '0' <= change[s[i] - '0']; ++i) {
          s[i] = (char) (change[s[i] - '0'] + '0');
        }
        break;
      }
    }
    return String.valueOf(s);
  }
}

class Solution {
public:
  string maximumNumber(string num, vector<int>& change) {
    int n = num.size();
    for (int i = 0; i < n; ++i) {
      if (change[num[i] - '0'] > num[i] - '0') {
        for (; i < n && change[num[i] - '0'] >= num[i] - '0'; ++i) {
          num[i] = change[num[i] - '0'] + '0';
        }
        break;
      }
    }
    return num;
  }
};

func maximumNumber(num string, change []int) string {
  s := []byte(num)
  for i, c := range num {
    if change[c-'0'] > int(c-'0') {
      for ; i < len(s) && change[s[i]-'0'] >= int(s[i]-'0'); i++ {
        s[i] = byte(change[s[i]-'0']) + '0'
      }
      break
    }
  }
  return string(s)
}