[swexpert] 1961. 숫자 배열 회전 (python)

728x90
반응형

zip은 배열의 가로 세로 배열을 바꿔주고 *은 리스트 내부 값을 가져온다

*[(1,2,3)] => (1,2,3) 이렇게 된다

 

t=int(input())

for i in range(1,t+1):
    n=int(input())
    matrix=[input().strip().split() for _ in range(n)] # 배열의 값 입력 받음
    result=[[] for _ in range(3)] # 결과(리턴 값)
    for idx in range(3):
        new_arr=[] # 뒤집을 배열 
        for z in zip(*matrix): #배열의 행열 뒤집기 
            new_arr.append(list(reversed(z))) #뒤집은 행을 넣어준다 
            result[idx].append(''.join(list(reversed(z)))) 
        matrix=new_arr
    print(f'#{i}')
    # 결과 배열의 행열을 한 번더 뒤집어줘야 된다
    for r in zip(*result):
        print(' '.join(r))


728x90
반응형
TAGS.

Comments