def startup(): pi, ni = map(int, input().split(' ')) if pi == 1: pyramidOne(ni) elif pi == 2: pyramidTwo(ni) def pyramidOne(n): i = 0 for x in range(1,n+1): if i == 0 and x == 1: if i !=1: i = i + 1 if i>0 and i<=2 and x<=4: if i == 1 and x == 4: i = i + 1 if i>1 and i<=3 and x<14: if i != 3 and x == 14: i = i + 1 print(i) startup() def pyramidTwo(n): i = 0 for x in range(1,n+1): if i == 0 and x == 1: if i !=1: i = i +1 if i>0 and i<=2 and x<=10: if i == 1 and x == 9: i = i + 1 if i>1 and i<=3 and x<=35: if i != 3 and x == 35: i = i + 1 print(i) startup() startup()
1 5 2 9 2 100
2 2 3
Traceback (most recent call last): File "./prog.py", line 37, in <module> File "./prog.py", line 4, in startup File "./prog.py", line 21, in pyramidOne File "./prog.py", line 6, in startup File "./prog.py", line 36, in pyramidTwo File "./prog.py", line 6, in startup File "./prog.py", line 36, in pyramidTwo File "./prog.py", line 2, in startup EOFError: EOF when reading a line