[알고리즘] 모래시계

📌

현수는 곳감을 만들기 위해 감을 깍아 마당에 말리고 있습니다. 현수의 마당은 N*N 격자판으 로 이루어져 있으며, 현수는 각 격자단위로 말리는 감의 수를 정합니다. 그런데 해의 위치에 따라 특정위치의 감은 잘 마르지 않습니다. 그래서 현수는 격자의 행을 기준으로 왼쪽, 또는 오른쪽으로 회전시켜 위치를 변경해 모든 감이 잘 마르게 합니다. 만약 회전명령 정보가 2 0 3이면 2번째 행을 왼쪽으로 3만큼 아래 그림처럼 회전시키는 명령 입니다.

 

index 0: 행

index 1: 0(왼쪽) 1(오른쪽)

index 2: 이동 횟수

 

ex ) 203 일 경우,

12 39 30 23 11 행을 왼쪽으로 3만큼 회전한다. 

23 11 12 39 30이 된다.

 

📌 input

5

10 13 10 12 15

12 39 30 23 11

11 25 50 53 15

19 27 29 37 27

19 13 30 13 19

3

2 0 3

5 1 2

3 1 4

 

📌 output

총 감의 개수

 

 

✔️ 내가 짠 코드

 

import sys
sys.stdin=open("input.txt", "r")

#2차원 행렬 (곶감)
n=int(input())
N=[list(map(int, input().split())) for _ in range(n)]
#회전명령
m=int(input())
M=[list(map(int, input().split())) for _ in range(m)]

for i in range(m):
    if M[i][1]==0: #왼쪽
        #회전하여 왼쪽으로 이동할 값+회전하여 기존 왼쪽에 있었던 값
        N[M[i][0]-1] = N[M[i][0]-1][M[i][2]:]+N[M[i][0]-1][:M[i][2]]
        #print(N)
    if M[i][1]==1:#오른쪽
    	#회전하여 기존 오른쪽에 있었던 값+회전하여 오른쪽으로 이동할 값
        N[M[i][0]-1] = N[M[i][0]-1][-M[i][2]:]+N[M[i][0]-1][:-M[i][2]]
        #print(N)

#모래시계 모양으로 계산한다.
res=0
s=0
e=n
for i in range(n):
    for j in range(s, e): 
        res+=N[i][j]
    if i<n//2: #n//2 전까진 양쪽에서 1씩 줄어든다.
        s+=1
        e-=1
    else:
        s-=1
        e+=1
print(res)