본문 바로가기

파이썬 알고리즘 풀이

18) 곳감(모래시계)

문제

M개의 회전명령을 실행하고 난 후 아래와 같이 마당의 모래시계 모양의 영역에는 감이 총 몇 개가 있는지 출력하는 프로그램을 작성하기

 

조건

현수의 마당은 N*N 격자판으 로 이루어져 있으며, 현수는 각 격자단위로 말리는 감의 수를 정합니다.

 

그런데 해의 위치에 따라 특정위치의 감은 잘 마르지 않습니다.

 

그래서 현수는 격자의 행을 기준으로 왼쪽, 또는 오른쪽으로 회전시켜 위치를 변경해 모든 감이 잘 마르게 합니다.

 

만약 회전명령 정보가 2 0 3이면 2번째 행을 왼쪽으로 3만큼 아래 그림처럼 회전시키는 명령 입니다.

 

제한사항

첫 줄에 자연수 N(3<=N<=20) 이 주어며, N은 홀수입니다.

 

두 번째 줄부터 N줄에 걸쳐 각 줄에 N개의 자연수가 주어진다.

 

이 자연수는 각 격자안에 있는 감의 개수이며, 각 격자안의 감의 개수는 100을 넘지 않는다.

 

그 다음 줄에 회전명령의 개수인 M(1<=M<=10)이 주어지고, 그 다음 줄부터 M개의 회전명령 정보가 M줄에 걸쳐 주어집니다.

 

총 감의 개수를 출력합니다.

 

문제풀이

N=int(input())
a=[list(map(int, input().split())) for _ in range(N)]
M=int(input())

s=0
e=N-1
res=0

for i in range(M):
  h, t, k = map(int, input().split())
  
  if t==0:
    for _ in range(k):
      a[h-1].append(a[h-1].pop(0))
  else:
    for _ in range(k):
      a[h-1].insert(0, a[h-1].pop())

for i in range(N):
    for j in range(s, e+1):
        res+=a[i][j]
    if i<N//2:
        s+=1
        e-=1
    else:
        s-=1
        e+=1

print(res)

'파이썬 알고리즘 풀이' 카테고리의 다른 글

20) 스도쿠 검사  (0) 2021.10.22
19) 봉우리  (0) 2021.10.22
17) 사과나무(다이아몬드)  (0) 2021.10.22
16) 격자판 최대합  (0) 2021.10.22
15) 수들의 합  (0) 2021.10.20