[백준, 파이썬, 18311번] 왕복

 

풀이

 

• 왕복으로 범위가 지정되는 것이기에
• 아래와 같이 구현하여 풀이
• 실수해서 오답 제출한 부분으로는 
• print(i%n+1) 으로 해야하는데 print(i%5+1)로 제출했다는 점... 
• 초기 예제 값 조심하기!!

 

import sys

input = sys.stdin.readline

n,k = map(int,input().split())

li = list(map(int,input().rstrip('\n').split()))

ch = li + li[::-1]

rg = 0

for i in range(len(ch)):

    if rg<=k<rg+ch[i]:

        print(n*2-i) if i>=n else print(i%n+1)

        break

    rg += ch[i]