Хоча код на Python не напишу, але здається це досить проста задача:
* перший цикл з лічильником, в якому з кожною ітерацією додається одиниця, і перевіряється маркер завершення циклу (від другого циклу)
** другий цикл, в тілі першого цикла, за кожною своєю ітерацією додає елементи в масив, перераховує їх та порівнює цю цифру з n; причому кількість ітерацій другого циклу дорівнює цифрі лічильника першого циклу, і цю ж цифру додає в масив в якості елемента; якщо в черговій ітерації кількість елементів зрівняється з n, то відмічається маркер завершення першого циклу та здійснюється "досрочное" завершення другого цикла.