得到连续 K 个 1 的最少相邻交换次数

/*
 思路：
 1. 使用Java Stream来收集1的位置。
 2. 计算前缀和并使用Stream API来完成相关计算。
 3. 最后一步和普通Java代码相似，但实现方式不同。
*/

import java.util.*;
import java.util.stream.*;

public class Solution {
    public int minMoves(int[] nums, int k) {
        List<Integer> ones = IntStream.range(0, nums.length)
                                      .filter(i -> nums[i] == 1)
                                      .boxed()
                                      .collect(Collectors.toList());
        int[] prefixSum = new int[ones.size() + 1];
        IntStream.range(0, ones.size()).forEach(i -> prefixSum[i + 1] = prefixSum[i] + ones.get(i));
        return IntStream.range(0, ones.size() - k + 1)
                        .map(i -> {
                            int mid = i + k / 2;
                            return (1 - k % 2) * ones.get(mid) + (prefixSum[i + k] - prefixSum[mid + 1]) - (prefixSum[mid] - prefixSum[i]);
                        })
                        .min()
                        .orElse(0);
    }
}