[Python] 곳감(모래시계)

[문제]

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

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

---

[풀이]

n = int(input())
a = [list(map(int, input().split())) for _ in range(n)]
m = int(input())    # 회전 명령 갯수

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())

s=0
e=n
res=0
for i in range(n):
    for j in range(s,e):
        res += a[i][j]

    if i < n//2:
        s += 1
        e -= 1
    else:
        s -= 1
        e += 1
print(res)

---

[내용]

1. 건들지도 못한 문제...

 

2. m=int(input()) #m개의 회전명령
b=[list(map(int, input().split())) for _ in range (m)]
이렇게 받았었다.

그러나, 이 회전명령은 각각의 세 숫자들이 쓸모가 각기 있으니깐 저렇게 한꺼번에 받는 것보다
h, t, k=map(int,input().split())
이렇게 받는 것이 더욱 낫다

 

3. 리스트의 원소들을 슬라이딩 할 때는 pop, insert 기억하자

- pop: 리스트들 당겨서 사용 가능

ex) a = [0,1,2,3,4] / a.pop(0) => [1,2,3,4]

- pop()만 사용한다면 맨 뒤에 있는것만 꺼내오는 것. => append, insert 와 함께 사용 가능

- a.pop(): 함수의 인자로 0을 주면 0번 인덱스를 삭제하고, 하나씩 왼쪽으로 당긴다. 인자로 공백을 주면 가장 마지막 인덱스 삭제

- pop(0): 맨 앞의 값을 꺼내 옴.

- a.insert(index,value): 반드시 삽입할 위치를 명시해야 한다. index 위치에 삽입되면 그 오른쪽의 원소들이 하나씩 오른쪽으로 밀린다.

 

4. N*N 행렬에서 다이아몬드 모양 / 모래시계 모양 만들기

- 다이아몬드 모양: 중간에서 시작해서 왼쪽으로 빼주고 오른쪽으로는 더해줌 -> N//2 행을 넘어가면 반대로 왼쪽으로 더하고 오른쪽으로 빼준다.

- 모래시계 모양: 각각 처음과 끝에서 시작해서 왼쪽으로는 더해주고 오른쪽으로는 빼주다가 -> N//2 행을 넘어가면 반대로 왼쪽으로 빼고 오른쪽으로 더해준다.