其他分享
首页 > 其他分享> > Leetcode 556.下一个更大元素Ⅲ

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