[M模拟] lc5824. 子字符串突变后可能得到的最大整数(模拟+代码细节+周赛251_2)

Posted 2021-08-27 Ypuyu

文章目录

• • 1. 题目来源
  • 2. 题目解析

1. 题目来源

链接：5824. 子字符串突变后可能得到的最大整数

2. 题目解析

显然从高位到低位顺序判断，如果更换后当前位变大，那就直接更换。注意更换是连续的一段子串。

故可以先跳过高位不可更换的，再进行更换即可，思路和第一名差不多嘻嘻，不过人家是原地算法，我又另外开辟了空间Orz，也可以用 flag 标记连续子段。

---

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$

---

代码：

我的代码。分三段，第一段处理高位不变的，第二段处理变大的，第三段再处理末尾字符即可。

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

第一名的代码，思路一致，实现更加简洁：

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

---

坑神采用 flag 标记连续子段，也是原地直接改变。

当开始改变时，flag=1，代表改变开始，不发生改变时，这里是严格的 >，因为如果是 = 的话，也是可以算作连续子段的，因为后面的可能也会变大，保留后面作为连续子段的可能性。

当遇见连续子段，有遇到结尾的话，改变结束。

这个 flag 操作值得学习。

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