Ответ:

k = int(input())

input() считывает ввод, а int() преобразует его в целое число.