def algo(n,u): u=3 s=u for i in range(...): u=... s=... return (u)