HelloCho

[Python] #1351. Count Negative Numbers in a Sorted Matrix(easy) 본문

코테/LeetCode

[Python] #1351. Count Negative Numbers in a Sorted Matrix(easy)

choo2969 2020. 9. 29. 14:01

문제 : 

이 문제는 단순하게 2차원 배열안에서 음수의 개수를 count하는 문제이다. 

특징은 내림차순으로 이미 정렬이 되어있다는 것이다.

class Solution:
    def countNegatives(self, grid: List[List[int]]) -> int:
        cnt = 0 
        for i in range(len(grid)):
            for j in range(len(grid[0])):
                if grid[i][j] < 0 :
                    cnt += len(grid[0]) - (j)
                    break
        return cnt

단순하게 접근했다. 2중 for문을 이용해서 배열 원소에 접근한다. 

만약 음수를 만난다면 그 행의 남은 원소역시 음수이기 때문에 행의 개수 - index를 이용해서 개수를 더해줬다. 

결과

'코테 > LeetCode' 카테고리의 다른 글

[Python] #125. Valid Palindrome(easy)  (0) 2020.09.30
[Python] #204. Count Primes(easy)  (0) 2020.09.30
[Python] #209. Minimum Size Subarray Sum(Medium)  (0) 2020.09.29
[Python] #36. Valid Sudoku(Medium)  (0) 2020.09.17
[Python] #18. 4Sum(Medium)  (0) 2020.09.15
Comments