题目描述
给定一个非负整数,你至多可以交换一次数字中的任意两位。返回你能得到的最大值。
示例 1 :
输入: 2736
输出: 7236
解释: 交换数字2和数字7。
示例 2 :
输入: 9973
输出: 9973
解释: 不需要交换。
注意:
给定数字的范围是 [0, 108]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-swap
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
struct cmp{
bool operator () (pair<char,int>& a,pair<char,int>& b){
return a.first < b.first;
}
};
class Solution {
public:
int maximumSwap(int num) {
priority_queue<pair<char,int>,vector<pair<char,int>>,cmp> que;
string str = to_string(num);
int len = str.length(),i;
for(i=0;i<len;i++) que.push(make_pair(str[i],i));
i = 0;
while(i<len && str[i]==que.top().first){
i++;
que.pop();
}
if(i>=len) return num;
int firidx = i,firval = que.top().first;
while(!que.empty() && que.top().first == firval){
swap(str[firidx],str[que.top().second]);
num = max(num,atoi(str.c_str()));
swap(str[firidx],str[que.top().second]);
que.pop();
}
return num;
}
};