Data Structures & Algorithms

[Algorithms] Basic-곶감(모래시계)

숄구-ml 2022. 5. 14. 17:03

  • 첫번째로, 수의 오른쪽 or 왼쪽 방향 이동은 파이썬의 pop(), insert(), append() 를 이용한다

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import sys
 
= int(input())
ar = [list(map(int, input().split())) for _ in range(n)]
= int(input())
 
for i in range(m):
    col, LorR, val = map(int, input().split())
 
    for _ in range(val):
        if LorR == 0:
            #left
            ar[col-1].append(ar[col-1].pop(0))
        else:
            #right
            ar[col-1].insert(0, ar[col-1].pop())
 
 
s=0
e=n-1
fruits=0
for i in range(n):
    for j in range(s, e+1):
        fruits += ar[i][j]
    if i < n//2:
        s+=1
        e-=1
    else:
        s-=1
        e+=1
 
print(fruits)
cs
728x90