给出一个二进制数组 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;
}
};