现有一个黑白棋游戏,初始时给出一排棋子,记作数组 chess
,其中白色棋子记作 0
,黑色棋子记作 1
。用户可以每次交换 任意位置 的两颗棋子的位置。
为了使得所有黑色棋子相连,请返回最少需要交换多少次。
示例 1:
输入: chess = [1,0,1,0,1,0] 输出: 1 解释: 有四种可能的方法可以把所有的 1 组合在一起: [1,1,1,0,0,0],交换 1 次; [0,1,1,1,0,0],交换 2 次; [0,0,1,1,1,0],交换 1 次; [0,0,0,1,1,1],交换 2 次。 所以最少的交换次数为 1。
思路:滑动窗口,假设数组中 1的个数为cnt,那么若经过最少次数交换之后连续1的序列的起始位置为i,则结束位置为i+cnt-1;这段序列区间内的所有0都要被交换走,所以维护一个长度为cnt的窗口,窗口内0个数最少的窗口就是最后的窗口,0的个数就是最少的交换次数。对于0的个数统计可以使用前缀和统计,preSum[i]表示到下标i时出现的0的个数,那么窗口内出现的0的个数为preSum[i]-preSum[cnt];
代码:
class Solution {
public int minSwaps(int[] chess) {
int cnt = 0;
int preSum[] = new i