백준 9455 박스

https://www.acmicpc.net/problem/9455

9455번: 박스

 

t=int(input())
for _ in range(t):
    m,n=map(int,input().split())
    matrix=[list(map(int, input().split())) for i in range(m)]
    transposed_matrix=list(zip(*matrix))
    sum_distance_box=0
    for i in transposed_matrix:
        for j in range(1,m):
            if i[::-1][j]==1:
                sum_distance_box+=i[::-1][:j].count(0)
    print(sum_distance_box)

행렬로 보면 transpose해서 박스 앞 즉 list의 값이 1 인경우 그 앞의 0의 개수를 전부 세서 더하면 움직인 거리

If this is matrix, Transpose it
if the value of the list is 1, the distance moved by counting all the 0s in front of it

numpy 라이브러리를 안쓰고 matrix를 transpose 하려고 Line 5 처럼 구현했지만
transposed 된 행렬의 행 내부 순서가 반대라서 Line9, Line10처럼 [::-1]와 같이 구현

In this circumstances, Not allowed to use numpy library
To transpose matrix , I did it like line 5
But rows in transposed matrix is reversed
So like line9,10 using [::-1] reverse rows