You are viewing a single comment's thread. Return to all comments →
but this will not work when d > len(a)
Correct, I added this line first to account for cases when d > len(a).
d = d % len(a) if d > len(a) else d
You don't need the if statement here. It's the same as writing:
d = d % a
true, but the constraints say 1 <= d <= n :)