力扣 1536. 排布二进制网格的最少交换次数（Java）

题目：

给你一个 n x n 的二进制网格 grid，每一次操作中，你可以选择网格的 相邻两行 进行交换。

一个符合要求的网格需要满足主对角线以上的格子全部都是 0 。

请你返回使网格满足要求的最少操作次数，如果无法使网格符合要求，请你返回 -1 。

主对角线指的是从 (1, 1) 到 (n, n) 的这些格子。

示例 1：

输入：grid = [[0,0,1],[1,1,0],[1,0,0]]
输出：3
示例 2：

输入：grid = [[0,1,1,0],[0,1,1,0],[0,1,1,0],[0,1,1,0]]
输出：-1
解释：所有行都是一样的，交换相邻行无法使网格符合要求。
示例 3：

输入：grid = [[1,0,0],[1,1,0],[1,1,1]]
输出：0

提示：

n == grid.length
n == grid[i].length
1 <= n <= 200
grid[i][j] 要么是 0 要么是 1 。

思路：

利用贪心的思想，从上到下逐行实现，若当前处理行为第i行，则第0到i-1行已经处理好。按照题意，第i行满足的条件为末尾连续0的个数大于等于n-i-1。考虑到交换次数最少，我们贪心的选择离第i行最近的那一行。

为了确定每行连续0的个数，我们首先用O(n2)的操作预处理出每一行最后一个1所在的位置，记为pos[i]。

这样我们只需要从上到下逐行确定，对于第i行，只要找到第i到n-1行中使得pos[j]<=i成立的最近的那一行j，然后将这一行交换到第i行即可。

代码实现：

import java.util.Arrays;
import java.util.Scanner;

public class Helloworld {

    public int minSwaps(int[][] grid){
        int n = grid.length;
        int[] pos = new int[n];
        Arrays.fill(pos,-1);
        for (int i = 0; i < n; i++) {
            for (int j = n-1; j >= 0 ; --j) {
                if(grid[i][j] == 1){
                    pos[i] = j;
                    break;
                }
            }
        }

        int ans = 0;
        for (int i = 0; i < n; i++) {
            int k = -1;
            for (int j = i; j < n; j++) {
                if(pos[j] <= i){
                    ans += j-i;
                    k = j;
                    break;
                }
            }

            if(k >= 0){
                for (int j = k;j > i;j--){
                    int temp = pos[j];
                    pos[j] = pos[j-1];
                    pos[j-1] = temp;
                }
            }else {
                return -1;
            }

        }

        return ans;
    }

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        int[][] grid = new int[n][n];
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                grid[i][j] = scanner.nextInt();
                System.out.print(grid[i][j]+",");
            }
            System.out.println("");
        }
        int ans = new Helloworld().minSwaps(grid);
        System.out.println(ans);

    }
}