670. Maximum Swap
Given a non-negative integer, you could swap two digits at most once to get the maximum valued number. Return the maximum valued number you could get.
Example 1:
Input: 2736
Output: 7236
Explanation: Swap the number 2 and the number 7.
Example 2:
Input: 9973
Output: 9973
Explanation: No swap.
Note:
- The given number is in the range [0, 10^8]
方法1:
grandyang: https://www.cnblogs.com/grandyang/p/7583875.html
思路:
先从右至左遍历第一次,记录下每个数位右边的最大值,储存成一个vector或者string。遍历第二次,对每个数位进行比较,如果当前位置不是右边的最大值,说明右边有数字需要被换过来,但是同样的最大值有可能有多个,我们要换的一定是最右边的最大值。所以开始第三次遍历,从右至左,找到这个最大值的位置,swap。
Complexity
Time complexity: O(n)
Space complexity: O(n)
class Solution {
public:
int maximumSwap(int num) {
int tmp = num;
int mx = INT_MIN;
vector<int> mx_vec;
vector<int> digit;
while (tmp > 0) {
digit.push_back(tmp % 10);
mx = max(mx, tmp % 10);
mx_vec.push_back(mx);
tmp /= 10;
}
reverse(mx_vec.begin(), mx_vec.end());
reverse(digit.begin(), digit.end());
int n = mx_vec.size();
int div = pow(10, n - 1);
for (int i = 0; i < n; i++) {
if (digit[i] < mx_vec[i]) {
int j = n - 1;
while (digit[j] != mx_vec[i]) j--;
swap(digit[i], digit[j]);
break;
}
}
int result = 0;
for (int a: digit) result = result * 10 + a;
return result;
}
};
方法2: greedy
官方题解:https://leetcode.com/problems/maximum-swap/solution/
思路:
类似于桶排的思路,首先遍历第一次,记录0 - 9每个digit最后一次出现的位置。因为根据上面的方法我们知道,需要交换的是第一次看到的小于右边最小值的那一位。所以第二次遍历的时候,对于每一位数字,我们只在last中所有大于它的那些中找,如果其所在数位在更右边,可以直接交换并返回。
Complexity
Time complexity: O(n)
Space complexity: O(n)
class Solution {
public:
int maximumSwap(int num) {
string s = to_string(num);
vector<int> last(10, 0);
for (int i = 0; i < s.size(); i++) {
last[s[i] - '0'] = i;
}
for (int i = 0; i < s.size(); i++) {
for (int d = 9; d > s[i] - '0'; d--) {
if (last[d] > i) {
char tmp = s[i];
s[i] = s[last[d]];
s[last[d]] = tmp;
return stoi(s);
}
}
}
return num;
}
};