iT邦幫忙

2021 iThome 鐵人賽

DAY 28
0
AI & Data

想到甚麼打甚麼系列 第 28

找LeetCode上簡單的題目來撐過30天啦(DAY28)

題號:73 標題:Set Matrix Zeroes 難度:Medium

Given an m x n integer matrix matrix, if an element is 0, set its entire row and column to 0's, and return the matrix.
You must do it in place.

Example 1:

Input: matrix = [[1,1,1],[1,0,1],[1,1,1]]
Output: [[1,0,1],[0,0,0],[1,0,1]]


Example 2:

Input: matrix = [[0,1,2,0],[3,4,5,2],[1,3,1,5]]
Output: [[0,0,0,0],[0,4,5,0],[0,3,1,0]]


Constraints:
• m == matrix.length
• n == matrix[0].length
• 1 <= m, n <= 200
• -231 <= matrix[i][j] <= 231 - 1

Follow up:
• A straightforward solution using O(mn) space is probably a bad idea.
• A simple improvement uses O(m + n) space, but still not the best solution.
• Could you devise a constant space solution?


我的程式碼

class Solution {
    public void setZeroes(int[][] matrix) {
        int i,j=0,k=0,size=(2*matrix[0].length*matrix.length) ;
        int[] temp = new int[size];
        Arrays.fill(temp, -1);
        for(i=0;i<matrix.length;i++){
            for(j=0;j<matrix[0].length;j++){
                if(matrix[i][j]==0){
                    temp[k] = i;
                    temp[k+1]=j;
                    k=k+2;
                }
            }
        }
        for(i=0;i<size;i=i+2){     
            if(temp[i]<0){
                break;
            }else{
                for(j=0;j<matrix[0].length;j++){
                    matrix[temp[i]][j] = 0;
                }
                for(j=0;j<matrix.length;j++){
                    matrix[j][temp[i+1]] = 0;
                }
            }
        }
    }
}

思考邏輯
本來以為只有0跟1想說先把要改成0的行跟列全部先存2,這樣就不會因為後面還有0出現而影響判斷,但後來發現陣列中的每個值可以是整數的整個範圍,因此作罷。
但偷懶如我,又想到in-place,代表可以宣告一維,於是宣告一個可以儲存mn2大小的一維,掃到0的座標存起來,兩兩一組,最後再把一維陣列中的座標對應的行跟列轉成0就完成拉

day28心得
原本以為我這個宣告一個陣列是做壞事,但我看解答邏輯也差不多嘛,心安理得了!


上一篇
找LeetCode上簡單的題目來撐過30天啦(DAY27)
下一篇
找LeetCode上簡單的題目來撐過30天啦(DAY29)
系列文
想到甚麼打甚麼30

1 則留言

0
juck30808
iT邦新手 3 級 ‧ 2021-10-14 12:10:06

恭喜即將邁入完賽~/images/emoticon/emoticon08.gif

我要留言

立即登入留言