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
n = int(input())
ar = [list(map(int, input().split())) for _ in range(n)]
m = 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