[파이썬 알고리즘] 곳감(모래시계)

곳감(모래시계)

📄 문제

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

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

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

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

▣ 입력예제
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

▣ 출력예제
362

🧞 풀이

풀이

'''
예를 들어, 2 0 3
0 1 2 3 4
3 4 0 1 2

3 -> 3 - 3 = 0
4 -> 4 - 3 = 0
2 -> 2 - 3 = -1 -> 4 == len() + (2 - 3)
1 -> 1 - 3 = -2 -> 3 == len() + (1 - 3)
0 -> 0 - 3 = -3 -> 2 == len() + (0 - 3)

한편, 2 1 3
0 1 2 3 4
2 3 4 0 1

0 -> 0 + 3 = 3
1 -> 1 + 3 = 4

2 -> 2 + 3 = 5 -> 1 = 5 - 4
3 -> 3 + 3 = 6 -> 2 = 6 - 4
4 -> 4 + 3 = 7 -> 3 = 7 - 4
'''

# 입력: 첫 줄 - N, 둘 째줄부터 N째줄까지 자연수 N개, 이후 M, M줄의 회전 명령
# 출력: 총 감의 개수

n = int(input())                                        # 정수 n 입력 및 할당
array = []                                              # 빈 array 할당
for i in range(n):                                      # n만큼 반복
    array.append(list(map(int,input().split())))        # array에 숫자 입력하여 추가

m = int(input())                                        # 정수 m 입력 및 할당
for i in range(m):                                      # m만큼 반복
    a, b, c = map(int, input().split())                 # a, b, c 입력 및 할당
    if b == 0:                                          # b가 0이면 왼쪽으로 회전
        for _ in range(c):                              # c만큼 반복
            array[a-1].append(array[a-1][0])            # 제일 왼쪽의 숫자를 제일 오른쪽에 추가
            array[a-1].pop(0)                           # 제일 왼쪽의 숫자 pop()으로 제거

    else:                                               # b가 1이면 오른쪽으로 회전
        for _ in range(c):                              # c만큼 반복
            array[a-1].insert(0, array[a-1][-1])        # 제일 오른쪽의 숫자를 제일 왼쪽에 추가
            array[a-1].pop()                            # 제일 오른쪽의 숫자 pop()으로 제거

s = 0                                                   # 시작 포인터: 0으로 할당
e = n                                                   # 끝 포인터: n으로 할당 (슬라이싱을 위해 n-1이 아닌, n)

sum_result = 0                                          # 합을 담을 sum_result 변수 0으로 할당

for j in range(n):                                      # n만큼 반복
    sum_result += sum(array[j][s:e])                    # 해당 조건의 슬라이싱된 리스트의 합을 sum_result에 더해줌
    if j < n // 2:                                      # 절반을 기준으로 위에 있는 행들의 경우,
        s += 1                                          # s는 1씩 커지고
        e -= 1                                          # e는 1씩 작아짐
    else:                                               # 절반을 기준으로 밑에 있는 행들의 경우,
        s -= 1                                          # s는 1씩 작아지고
        e += 1                                          # e는 1씩 커짐

print(sum_result)                                       # 합 출력

 

다른 사람 풀이

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

res = 0
s = 0
e = n - 1
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)