LeetCode 2938. 区分黑球与白球（贪心、归并排序）

最新推荐文章于 2024-07-18 01:42:44 发布

岁忧

最新推荐文章于 2024-07-18 01:42:44 发布

阅读量292

点赞数 5

分类专栏： LeetCode 文章标签： leetcode 算法职场和发展 c++ 数据结构归并排序贪心

本文链接：https://blog.csdn.net/weixin_46028214/article/details/139494407

版权

LeetCode 专栏收录该内容

362 篇文章 0 订阅

订阅专栏

2938. 区分黑球与白球

在这里插入图片描述

思路：方法一，最小的移动次数，归并排序。

class Solution {
public:
    long long cnt=0;
    void merge_sort(int l,int r,string &s){
        if(l>=r) return;
        int mid=(l+r)/2;
        merge_sort(l,mid,s);
        merge_sort(mid+1,r,s);
        string tmp="";
        int i=l,j=mid+1;
        while(i<=mid&&j<=r){
            if(s[i]<=s[j]){
                tmp.push_back(s[i++]);
            }else{
                cnt+=mid-i+1;
                tmp.push_back(s[j++]);
            }
        }
        while(i<=mid){
            tmp.push_back(s[i++]);
        }
        while(j<=r){
            tmp.push_back(s[j++]);
        }
        for(int i=l;i<=r;i++){
            s[i]=tmp[i-l];
        }
    }
    long long minimumSteps(string s) {
        merge_sort(0,s.size()-1,s);
        return cnt;
    }
};

思路：方法二，贪心。遍历字符串s时，只需要记录已经遍历过的部分出现1的次数cnt，然后遇到0，就将sum+=cnt。理由，每一个0肯定要和它前面的1进行互换。

class Solution {
public:
    long long minimumSteps(string s) {
        long long sum=0;
        int cnt=0;
        for(int i=0;i<s.size();i++){
            if(s[i]=='1') cnt++;
            else sum+=cnt;
        }
        return sum;
    }
};