CSES - Kuningattaret

Tehtäväsi on laskea, monellako tavalla n \times n shakkilaudalle voidaan sijoittaa k kuningatarta niin, että mitkään kaksi kuningatarta eivät uhkaa toisiaan.

Voit olettaa, että 1 \le n \le 8 ja 1 \le k \le 3. Ratkaisusi tulee toimia tehokkaasti kaikissa näissä tapauksissa.

Toteuta tiedostoon queens.py funktio count, joka laskee tuloksen.

def count(n, k):
    # TODO

if __name__ == "__main__":
    print(count(2, 1)) # 4
    print(count(2, 2)) # 0
    print(count(5, 3)) # 204
    print(count(7, 1)) # 49
    print(count(7, 2)) # 700
    print(count(7, 3)) # 3628