yrOcrE - Online C++0x Compiler & Debugging Tool

#include <bits/stdc++.h>
using namespace std;

const int N=10;
int n, k, ans;
bool check[N][N], col[N];

int di[8]={-2, -2, -1, 1, 2, 2, -1, 1};
int dj[8]={-1, 1, 2, 2, -1, 1, -2, -2};
bool inside(int i, int j) //hàm kiểm tra ô (i, j) có nằm trong bàn cờ hay không
{
    return i>0 && i<=n && j>0 && j<=n;
}
int cnt;
void btrack(int i)
{
    if(i>n){
        if(cnt==k) ++ans;
        return;
    }
    if(cnt<k){ //trường hợp đặt tại hàng i
        for(int j=1;j<=n;++j){
            if(col[j] || check[i][j]) continue;

            ++cnt;
            col[j]=1;
            for(int x=0;x<8;++x){
                int ni=i+di[x], nj=j+dj[x]; //vị trí từ ô (i, j) có thể di tới theo kiểu quân Mã
                if(inside(ni, nj)) check[ni][nj]=1;
            }

            btrack(i+1);

            --cnt;
            col[j]=0;
            for(int x=0;x<8;++x){
                int ni=i+di[x], nj=j+dj[x];
                if(inside(ni, nj)) check[ni][nj]=0;
            }
        }
    }
    btrack(i+1); //trường hợp không đặt tại hàng i
}

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);

    cin >> n >> k;
    btrack(1);
    cout << ans;
    return 0;
}