2010 플러그

📖 question

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

2010번: 플러그

 

✍️ answer

n = int(input())
result = 0

for _ in range(n):
    p = int(input())
    result += p
result -= (n-1)  # 연결되는 코드는 하나씩 빼줘야하는데 마지막은 연결이 필요없으므로 -1빼줌
print(result)

파이썬3에서 위의 소스로 제출하면 시간 초과가 되므로 pypy3로 제출했다.

시간 초과가 일어나는 이유는 반복문에서 input()으로 입력받았기 때문인데,

이를 해결하기 위해 sys.stdin.readline()을 사용해야 한다.

import sys
n = int(sys.stdin.readline())
sum = 0
for i in range(n):
    sum += int(sys.stdin.readline())
print(sum - (n - 1))

 

참고

2010 플러그 문제풀이

sys.stdin.readline() 입력받기