вход с разделителями в Python

Question

У меня есть проблема со следующей спецификации:

Входной сигнал:

Первая строка содержит целое число N, количество элементов в данной сходящейся последовательности. Далее следует N целых чисел A1, A2 .... An, Ai - i-й элемент данной последовательности. Эти числа могут быть разделены пробелом или разделителем новой строки.

Как я могу обрабатывать входные данные таким образом? Я попытался следующим, но это работает только для пространства отделившегося elements.When используется с новой строкой разделенных элементов грейдер бросает «превышено время»

import sys 
counter=0 
A=[] 
for line in sys.stdin: 
    if counter!=0: 
     A+=[int(Ai) for Ai in line.split()] 
    else: 
     N=int(line) 
     counter+=1 

Answer 1

Предполагая, что вы читаете из файла (sys.stdin также файл):

f = open('path/to/file') 
N = int(f.readline().strip()) 
A = [] 
while len(A) < N: 
    A.extend(int(i) for i in f.readline().strip().split()) 

Надеется, что это помогает