LeetCode 1505. 最多 K 次交换相邻数位后得到的最小整数(思维+树状数组)

题意:
给你一个字符串 num 和一个整数 k 。
其中,num 表示一个很大的整数,字符串中的每个字符依次对应整数上的各个 数位 。

你可以交换这个整数相邻数位的数字 最多 k 次。

请你返回你能得到的最小整数,并以字符串形式返回。

数据范围:
1 <= num.length <= 30000
num 只包含 数字 且不含有 前导 01 <= k <= 10^9
解法:
由于操作之后数位个数不会发生改变,
因此要求整数最小其实就是要求字典序最小.

字典序最小显然要先让最高位尽可能小,
对于答案的第一位(左边),从数字0开始枚举,判断是否能在k步之内移动一个最近的过来.
假设将pos位置上的数移动过来了,那么最高位到pos之间的所有数向右移动了一位,
那么对于每个字符s[i],移动之后的位置应该是:i+([i+1,n]移动到左边的字符数量).

([i+1,n]移动到左边的字符数量)可以用树状数组维护,具体做法是:
当pos位置的数被移动,令c[pos]]=1,
那么我们统计[i+1,n]中有多少个数移动到左边,
其实就是计算[i+1,n]1的个数即区间和,可以用树状数组计算.
code:
struct BIT{
    const static int maxm=3e4+5;
    int c[maxm];
    int lowbit(int i){
        return i&-i;
    }
    void add(int i,int t){
        while(i<maxm)c[i]+=t,i+=lowbit(i);
    }
    int ask(int i){
        int ans=0;
        while(i)ans+=c[i],i-=lowbit(i);
        return ans;
    }
    void init(){
        memset(c,0,sizeof c);
    }
};
class Solution {
public:
    BIT T;
    string minInteger(string s, int k) {
        T.init();
        string ans;
        int n=s.size();
        vector<queue<int> >pos(10);
        for(int i=0;i<n;i++){
            pos[s[i]-'0'].push(i+1);//下标要变为[1,n],因为BIT不能用0.
        }
        for(int i=1;i<=n;i++){
            for(int j=0;j<10;j++){
                if(pos[j].size()){
                    int dif=T.ask(n)-T.ask(pos[j].front());
                    int dist=pos[j].front()+dif-i;
                    if(dist<=k){
                        T.add(pos[j].front(),1);
                        pos[j].pop();
                        ans+=(char)(j+'0');
                        k-=dist;
                        break;
                    }
                }
            }
        }
        return ans;
    }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值