Недавно начал изучать питон. Сейчас решил попробовать себя в олимпиаде. В одной задаче написано, что сначала будет вводиться количество папок, а потом количество файлов в этих папках. И вот возникает вопрос, как считать число файлов, которое всегда будет меняться? Если начинать через If A = int(input()) (количество папок)
if A == 1:
M = int(input()) (сколько файлов в одной папке)
if A == 2:
M,N= map(int,input().split()) (сколько файлов в каждой папке отдельно)