【leetcode-标记法】找到所有数组中消失的数字/矩阵置零

找到所有数组中消失的数字

给定一个范围在 1 ≤ a[i] ≤ n ( n = 数组大小 ) 的 整型数组，数组中的元素一些出现了两次，另一些只出现一次。
找到所有在 [1, n] 范围之间没有出现在数组中的数字。
您能在不使用额外空间且时间复杂度为O(n)的情况下完成这个任务吗? 你可以假定返回的数组不算在额外空间内。

示例:
输入:
[4,3,2,7,8,2,3,1]
输出:
[5,6]

标记法

class Solution {
    public List<Integer> findDisappearedNumbers(int[] nums) {
        int len = nums.length;
        for (int i = 0; i < len; i++) 
            nums[(nums[i] - 1) % len] += len;

        List<Integer> result = new ArrayList<>();
        for (int j = 0; j < len; j++)
            if (nums[j] <= len)
                result.add(j + 1);
        return result;
    }
}

排序法

class Solution {
    public List<Integer> findDisappearedNumbers(int[] nums) {
        int i = 0, len = nums.length;
        while (i < len) {
            if (nums[nums[i] - 1] == nums[i] || nums[i] == i + 1) {
                i++;
                continue;
            }  
            int tmp = nums[nums[i] - 1];
            nums[nums[i] - 1] = nums[i];
            nums[i] = tmp;
        }
        
        List<Integer> result = new ArrayList<>();
        for (int j = 0; j < len; j++)
            if (j + 1 != nums[j])
                result.add(j + 1);
        return result;
    }
}

矩阵置零

给定一个 m x n 的矩阵，如果一个元素为 0，则将其所在行和列的所有元素都设为 0。请使用原地算法。

示例 1:
输入:
[
[1,1,1],
[1,0,1],
[1,1,1]
]
输出:
[
[1,0,1],
[0,0,0],
[1,0,1]
]

示例 2:
输入:
[
[0,1,2,0],
[3,4,5,2],
[1,3,1,5]
]
输出:
[
[0,0,0,0],
[0,4,5,0],
[0,3,1,0]
]

额外存储空间法

记录下数值为零的格子的行号和列号。时间复杂度：$O(M\times N)$，空间复杂度：$O(M+N)$。

class Solution {
    public void setZeroes(int[][] matrix) {
        int m = matrix.length, n = matrix[0].length;
        Set<Integer> rows = new HashSet<>();
        Set<Integer> columns = new HashSet<>();
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (matrix[i][j] == 0) {
                    rows.add(i);
                    columns.add(j);
                }
            }
        }

        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (rows.contains(i) || columns.contains(j))
                    matrix[i][j] = 0;
            }
        }

        
    }
}

标记法

将0元素所在行和列的第一个元素设置为0，再次遍历时检查第一个元素。
时间复杂度：$O(M\times N)$，空间复杂度：$O(1)$。

class Solution {
    public void setZeroes(int[][] matrix) {
        int m = matrix.length, n = matrix[0].length;
        boolean firstRow = false, firstCol = false;
        for (int i = 0; i < m; i++) {
            if (matrix[i][0] == 0) {
                firstCol = true;
                break;
            }
        }
        for (int j = 0; j < n; j++) {
            if (matrix[0][j] == 0) {
                firstRow = true;
                break;
            }
        }
        for (int i = 1; i < m; i++) {
            for (int j = 1; j < n; j++) {
                if (matrix[i][j] == 0) {
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        } 

        for (int i = 1; i < m; i++) {
            if (matrix[i][0] == 0) {
                for (int j = 0; j < n; j++)
                    matrix[i][j] = 0;
            }
        }
        for (int j = 1; j < n; j++) {
            if (matrix[0][j] == 0) {
                for (int i = 0; i < m; i++)
                    matrix[i][j] = 0;
            }
        }

        if (firstRow) {
            for (int j = 0; j < n; j++) 
                matrix[0][j] = 0;
        }
        if (firstCol) {
            for (int i = 0; i < m; i++) 
                matrix[i][0] = 0;
        }
    }
}