백준 알고리즘 1247번 부호(python)

단순 사칙연산 문제이다. 

for _ in range(3):
    N = int(input())
    li = [int(input()) for i in range(N)]
    if sum(li) == 0:
        print("0")
    elif sum(li) > 0:
        print("+")
    else:
        print("-")

위의 코드가 맞긴 하는데 그냥 제출하면 시간 초과가 나온다.

stdin.readline를 사용해서 입력을 빠르게 받아오면 해결된다.

from sys import stdin

for _ in range(3):
    N = int(stdin.readline())
    li = [int(stdin.readline()) for i in range(N)]
    if sum(li) == 0:
        print("0")
    elif sum(li) > 0:
        print("+")
    else:
        print("-")