排布二进制网格的最少交换次数

/* 思路：
1. 使用流操作找到每行最右侧的1的位置。
2. 逐行检查是否满足条件，不满足则寻找下一行并交换。
3. 统计交换次数。
*/

import java.util.stream.IntStream;

public class Solution {
    public int minSwaps(int[][] grid) {
        int n = grid.length;
        int[] maxRightOnes = IntStream.range(0, n)
            .map(i -> findRightmostOne(grid[i]))
            .toArray();
        
        int swaps = 0;
        for (int i = 0; i < n; i++) {
            if (maxRightOnes[i] < i) {
                int j = IntStream.range(i, n)
                    .filter(k -> maxRightOnes[k] >= i)
                    .findFirst().orElse(-1);
                if (j == -1) return -1;
                
                for (int k = j; k > i; k--) {
                    swap(maxRightOnes, k, k - 1);
                    swaps++;
                }
            }
        }
        return swaps;
    }

    private int findRightmostOne(int[] row) {
        return IntStream.range(0, row.length)
            .map(i -> row[row.length - 1 - i] == 1 ? row.length - 1 - i : -1)
            .filter(i -> i != -1)
            .findFirst().orElse(-1);
    }

    private void swap(int[] arr, int i, int j) {
        int temp = arr[i];
        arr[i] = arr[j];
        arr[j] = temp;
    }
}