Skip to content
本页目录

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

https://leetcode.cn/problems/minimum-swaps-to-group-all-1s-together

给出一个二进制数组 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个个数,设置滑动大小 计算滑动窗口内0的个数,取最小值

参考代码

csharp
public class Solution {
    public int MinSwaps(int[] data) {
    	int count = 0;
    	for(int i=0; i<data.Length; i++){
    		if(data[i] == 1){
    			count++;
    		}
    	}
    	int left = 0;
    	int right = count-1;
    	//统计初始0的个数
    	int zeroCount = 0;
    	for(int i=0;i<=right;i++){
    		if(data[i] == 0){
    			zeroCount++;
    		}
    	}
    	int minCount = zeroCount;
        left++;
        right++;
    	while(right < data.Length){
    		if(data[left-1] == 0){
    			zeroCount--;
    		}
    		if(data[right] == 0){
    			zeroCount++;
    		}
    		minCount = Math.Min(zeroCount,minCount);            
            left++;
    		right++;
    	}
    	return minCount;
    }
}

Released under the MIT License.