Leetcode 556.下一个更大元素Ⅲ
作者:互联网
一道比较显然的贪心。
首先我们很容易想到枚举所有的排列情况,但是这样是显然不能通过的。其次我们可以贪心地考虑,思路:假设我们有一个数$n$是ABCDEF(ABCDEF都是数字),既然要找大于$n$的最小排列,我们其实只需要从个位开始考虑,变化越小的数位越好。
以数$1961283241$为例,从个位的1开始考虑与哪个数位交换,因为这组数字中没有比1更小的0,1已经是最小的,所以它无论与之前的什么数字交换,都会导致新数比原数更大,所以我们跳过1考虑十位上的4,会立即发现,交换十位4与百位2可以得到一个更大的新数$1961283421$!但是这还没有得到结果,得到这个新数,只代表我们找到了一个可能的结果——我们需要寻找的是最小的大于原数的数。我们得到的这个新数只是比原数大,可能有比这个新数更小的合法结果。怎么找到更小的合法结果呢?我们只需要把4之后的数按照从小到大排序,保证小的数在大的数位上就行。大概是这种思路。
代码主体就是一次最优交换+片段排序。
最优交换我设低位序号为变量$from$,高位序号为变量$to$。一次合法的交换应该保证$num[from]>num[to]$(这样才能确保交换后得到的数大于原数)。最优交换的贪心原则:
1)$to$的序号应当尽可能地小。因为一次交换只能让$num[to]$增大!更大的数位增大会让结果更大,所以$to$的数位序号应当尽可能小。
2)在$to$的序号保持不变的情况下,交换前的$num[from]$应当尽可能小,因为$to$是发生变化的最大数位。
大概就是这样,找到最优交换后将$to$之前的小数位进行排序。
代码写出来后看起来效率还可以:
vector<int> num; bool cmp(int a,int b){ return a>b; } int Ans(const int lim){ if (lim>=10){ //十亿以上 long long ans=0; long long max=2147483647; long long key=1; for (int i=0;i<lim;i++) ans+=(long long)(num[i]*key),key*=(long long)10; if ((long long)(ans>max)) return -1; else return (int)ans; } int ans=0; for (int i=0,key=1;i<lim;i++) ans+=(num[i]*key),key*=10; return ans; } class Solution { public: int nextGreaterElement(int n) { num.clear(); while (n){ num.push_back(n%10); n/=10; } bool flag=false; int lim=num.size(); int from=-1,to=-1; for (int i=0;i<lim;i++){ for (int j=i+1;j<lim;j++) if (num[j]<num[i]){ flag=true; if (from==-1) from=i,to=j; else{ if (to>j) from=i,to=j; else if (to==j){ if (num[i]<num[from]) from=i; } } j=lim; } } if (!flag) return -1; int temp=num[from]; num[from]=num[to],num[to]=temp; sort(num.begin(),num.begin()+to,cmp); return Ans(lim); } };
标签:新数,556,int,元素,交换,long,num,Leetcode,数位 来源: https://www.cnblogs.com/wegret/p/16440144.html