1151. 最少交换次数来组合所有的 1

给出一个二进制数组 data,你需要通过交换位置,将数组中 任何位置 上的 1 组合到一起,并返回所有可能中所需 最少的交换次数。

示例 1:

输入:[1,0,1,0,1]
输出:1
解释: 
有三种可能的方法可以把所有的 1 组合在一起:
[1,1,1,0,0],交换 1 次;
[0,1,1,1,0],交换 2 次;
[0,0,1,1,1],交换 1 次。
所以最少的交换次数为 1。

示例 2:

输入:[0,0,0,1,0]
输出:0
解释: 
由于数组中只有一个 1,所以不需要交换。

示例 3:

输入:[1,0,1,0,1,0,0,1,1,0,1]
输出:3
解释:
交换 3 次,一种可行的只用 3 次交换的解决方案是 [0,0,0,0,0,1,1,1,1,1,1]。

提示:

1 <= data.length <= 10^5
0 <= data[i] <= 1

思路:考虑两种写法,借助滑动窗口处理。

第一种:首先统计出数组中1的个数,作为滑动窗口的长度,之后再原数组中找到某个窗口中包含1个数最多的窗口,窗口大小减最大值就是结果。
数组大小为1e5,所以处理滑动窗口的时候要注意下,用一个cnt变量统计每个窗口中1的个数,该窗口前一个值若为1,cnt–,该窗口后一个值若为1,cnt++,如此一直往后推即可。

class Solution {
public:
    int minSwaps(vector<int>& data) {
        int num1 = 0;
        for(int i = 0; i < data.size(); i++) {
            if(data[i] == 1) {
                num1++;
            }
        }
        int cnt = 0;
        for(int i = 0; i < num1; i++) {
            if(data[i] == 1) {
                cnt++;
            }
        }
        int maxm = cnt;
        
        for(int i = 1; i <= data.size() - num1; i++) {
            if(data[i-1] == 1) {
                cnt--;
            }
            if(data[i+num1-1] == 1) {
                cnt++;
            }
            maxm = max(maxm, cnt);
        }
        return num1 - maxm;
    }
};

第二种:更好的写法,参考了大佬的代码
大致上就是开一个s数组维护原数组的前缀和,然后借助前缀和来统计滑动窗口中1的个数,比上面写法更好

class Solution {
public:
    int minSwaps(vector<int>& data) {
        int n = data.size();
        vector<int> s(n+1);
        for(int i = 1; i <= n; i++) {
            s[i] = s[i-1] + data[i-1];
        }
        int m = s[n];
        int res = n;
        for(int i = m; i <= n; i++) {
            res = min(res, m - (s[i] - s[i-m]));
        }
        return res;
    }
};
  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值