def f(i,k): r=2-(i>k) if i<2:r=1 elif i