1. 题目来源
2. 题目解析
显然从高位到低位顺序判断,如果更换后当前位变大,那就直接更换。注意更换是连续的一段子串。
故可以先跳过高位不可更换的,再进行更换即可,思路和第一名差不多嘻嘻,不过人家是原地算法,我又另外开辟了空间Orz,也可以用 flag 标记连续子段。
- 时间复杂度: O ( n ) O(n) O(n)
- 空间复杂度: O ( n ) 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;
}
};