Set Matrix Zeros

박수빈·2022년 1월 17일
0

leetcode

목록 보기
5/51


문제

  • m*n 메트릭스
  • 하나가 0이면 그 row, col 0으로 만들어.
  • in place

풀이

  • in place이니까 dfs 하기 어려워
  • 첫 순회 시에 0의 인덱스 저장
  • 인덱스 이용해서 그 행과 열 값 변경
        nRow = len(matrix)
        nCol = len(matrix[0])
        # 0인 곳의 인덱스 찾기
        zeroIndices = []
        for row, li in enumerate(matrix):
            for col, el in enumerate(li):
                if el == 0:
                    zeroIndices.append([row, col])
        # 인덱스 이용해서 행과 열의 값 변경
        for row, col in zeroIndices:
            # 행 값 변경
            for c in range(nCol):
                matrix[row][c] = 0
            # 열 값 변경
            for r in range(nRow):
                matrix[r][col] = 0
        return None

결과

1트에 성공!

profile
개발자가 되고 싶은 학부생의 꼼지락 기록

0개의 댓글