给定一个非负整数,你至多可以交换一次数字中的任意两位。返回你能得到的最大值。
示例 1 :
输入: 2736 输出: 7236 解释: 交换数字2和数字7。
示例 2 :
输入: 9973 输出: 9973 解释: 不需要交换。
注意:
- 给定数字的范围是 [0, 108]
思路:这道题如果遍历每一种情况,时间复杂度是很高的,所以我们采用贪心算法,一次遍历O(n)即可解决问题。
基于贪心算法的基本观察是,从后向前遍历,如果遇到更大的值maxValue,那么保存这个值的下标(index)和值(value),如果在这个值的左边遇到比maxValue小的数,那么保存对应的值leftValue和leftIndex,这两个配对组成一组候选值。从右向左遍历数组,遇到满足上述条件的值就覆盖更新对应的值,最后交换leftValue和maxValue对应的下标即可。
参考代码:
class Solution {
public:
int maximumSwap(int num) {
int maxValue = -1, maxIndex = -1;
int leftIndex = -1, rightIndex = -1;
string num_str = to_string(num);
for (int i = num_str.size() - 1; i >= 0; i--) {
if (num_str[i] > maxValue) {
maxValue = num_str[i];
maxIndex = i;
continue;
}
if (num_str[i] < maxValue) {
leftIndex = i;
rightIndex = maxIndex;
}
}
if (leftIndex == -1) {
return num;
}
swap(num_str[leftIndex], num_str[rightIndex]);
return stoi(num_str);
}
};