[Python Algorithm] 탐색&시뮬레이션(2차원 리스트 탐색) - 곶감(모래시계)

> 탐색 & 시뮬레이션 (2차원 리스트 탐색 편)

 

📖 문제 : 곶감(모래시계)

 

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

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

첫 번째 수는 행번호, 두 번째 수는 방향인데 0이면 왼쪽, 1이면 오른쪽이고, 세 번째 수는 회 전하는 격자의 수입니다.
M개의 회전명령을 실행하고 난 후 아래와 같이 마당의 모래시계 모양의 영역에는 감 이 총 몇 개가 있는지 출력하는 프로그램을 작성하세요.

 

입력설명
첫 줄에 자연수 N(3<=N<=20) 이 주어며, N은 홀수입니다.
두 번째 줄부터 N줄에 걸쳐 각 줄에 N개의 자연수가 주어진다.
이 자연수는 각 격자안에 있는 감의 개수이며, 각 격자안의 감의 개수는 100을 넘지 않는다. 그 다음 줄에 회전명령의 개수인 M(1<=M<=10)이 주어지고, 그 다음 줄부터 M개의 회전명령 정보가 M줄에 걸쳐 주어집니다.

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

입력예제 1

5
10 13 10 12 15

25 19 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

출력예제 1
362

 

코드

import sys

n = int(input())
a = [list(map(int,input().split())) for _ in range(n)]
m = int(input())
for i in range(m):
    height, direction, count = map(int, input().split())
    if direction == 0:		# left
    	for _ in range(count):
    		a[height-1].append(a[height-1].pop(0))
    
    else :		# right  
    	for _ in range(count):
    		a[height-1].insert(0, a[height-1].pop())
        
s = 0
e = n-1
result = 0
for i in range(n):
	for j in range(s,e+1):
		result += a[i][j]
        if i < n//2:
		s += 1
		e -= 1
	else:
		s -= 1
		e += 1
print(result)

⭐️ Point ! ⭐️

• 그때 그때 써야하는 입력값들은 한번에 리스트로 저장보다는 for문을 돌리면서 입력 받기
• 왼쪽으로 회전해야 할 때, list[index].append(list[index].pop(0))  
  : pop(0)으로 0번째 인덱스 값을 빼주어 return된 값을 바로 append()함수를 사용하여 값 넣어주기
• 오른쪽으로 회전해야 할 때, list[index].insert(0, list[index].pop())
  : insert() 함수를 사용해서 0번째 인덱스 값에 pop() 함수를 사용하여 제일 끝 값을 return 받아 인덱스 0번에 넣어주기
• 그 전 게시물 '사과나무' 에서 응용하여 특정 규칙이 있는 이차원 배열의 값들을 더해주는 방법

 

 

출처

• 인프런 : 파이썬 알고리즘 문제 풀이