思路:滑动窗口
窗口的大小就是数组中1的个数,确定了窗口的大小,定义结果集为res,初始化窗口左右边界从数组开头开始,用sum记录窗口中0的个数,
因为第一个元素和最后一个元素相邻,所以我们可以再用一个数组nums拼接到数组后面,来完成环形数组,但在实际写代码中,我们只需要在滑动窗口的过程中让条件为窗口的右边界R<2*nums.size(),然后用nums[L%nums.size()]找到数组中的元素即可
在滑动窗口的过程中,如果sum<res,就更新结果集res=sum;左边界右移和右边界右移的同时更新窗口中0的个数sum,循环结束后,返回结果集res即可
代码:
class Solution {
public:
int minSwaps(vector<int>& nums) {
int n=nums.size();
int wsize=accumulate(nums.begin(),nums.end(),0);//用求和的方式得到数组中1的个数,窗口的大小
if(wsize==0) return {};//如果数组中没有1,返回空数组
int L=0,R=wsize;//滑动窗口边界左右边界,
int res=wsize;//初始化结果集,
int sum=wsize-accumulate(nums.begin(),nums.begin()+wsize,0);//初始窗口中0的个数
while(R<nums.size()*2){//R是窗口的右边界,R走完两个数组数组长度表示遍历完成
if(sum<res){//更新结果集
res=sum;
}
if(nums[L%n]==0){//左边界右移,同时更新窗口中0的个数
sum--;
}
L++;
if(nums[R%n]==0){//右边界右移,同时更新窗口中0的个数
sum++;
}
R++;
}
return res;
}
};