There is an error for test case k=2 and n=2

Describe your bug

here the test case is resulting output “no” for k=2 and n=2
but the solution shows output “yes”

Screenshot

Additional info

problem name(BALL AND BOXES)==the test case is error please check for input n=2 and k=2

![Screenshot (6)|690x388]