본문 바로가기
Beakjoon&프로그래머스/파이썬

[백준/파이썬] 9455번 박스

by 현장 2022. 2. 19.

-코드

for _ in range(int(input())):
    n, m = map(int, input().split())
    arr = [list(map(int, input().split())) for _ in range(n)]
    r = 0
    for i in range(m):
        arr2 = []
        for j in range(n):
            arr2.append(arr[j][i])
        for k in range(n):
            if arr2[k] == 1:
                r += arr2[k:].count(0)
    print(r)

처음에 계산을 하는 방법을 움직이면서 움직이는 횟수를 새는 것으로 생각을 했으나 다시 생각해보니 그러면 너무 복잡해져서 고민하다가 0의 개수로 하는 방법이 있지 않을까 생각해보고 생각한 것이 맞는지 찾아보면서 해결을 했습니다.