[백준/BOJ] 15738 뒤집기 (파이썬 Python)

 

https://www.acmicpc.net/problem/15738

 

어려운 PS 문제는 아니었고 왼쪽에서 또는 오른쪽에서 뒤집을 때 규칙만 잘 찾아내면 되는 문제이다. 

 

따지고 보면 N, K, M 과 i 값만 필요할 뿐 배열의 요소는 받고 전혀 활용되지 않기 때문에

 

input( )으로만 남겨두어도 무방할 것 같다.

 

import sys
input = sys.stdin.readline

# 입력
n, k, m = map(int, input().split())
arr = list(map(int, input().split()))

# 알고리즘
for _ in range(m):
    i = int(input())
    if i > 1 and k <= i:
        k = (i + 1) - k
    elif i < 0 and k >= n+i+1:
        k = 2 * (n + i + 1) - i - 1 - k
        
print(k)