Lähetit

#include <iostream>
#include <vector>
#include <map>
#include <string>
#include <algorithm>
#include <utility>

using namespace std;

typedef long long LL;

LL ans = 0;
LL n, k;
    
void solve(vector<vector<bool> > available, LL left){
    /*for(int i = 0; i < n; i++){
        for(int j = 0; j < n; j++){
            cout << available[i][j];
        }
        cout << endl;
    } cout << endl;*/

    if(left == 0){
//        cout << "++" << endl;

        ans++; return;
    }

    for(LL r = 0; r < n; r++){
        for(LL c = 0; c < n; c++){
            if(available[r][c]){
                vector<vector<bool> > available_new = available;
                LL r2 = r - c;
                for(LL c2 = 0; c2 < n; c2++, r2++){

                    if(r2 >= 0 && r2 < n)
                        available_new[r2][c2] = false;
                }
                r2 = r + c;
                for(LL c2 = 0; c2 < n; c2++, r2--){
                    if(r2 >= 0 && r2 < n)
                        available_new[r2][c2] = false;
                }
                solve(available_new, left - 1);
            }
        }
    }
}

LL factorial(LL x){
    LL res = 1;
    for(LL i = 1; i <= x; i++)
        res *= i;
    return res;
}

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

    cin >> n >> k;
    vector<vector<bool> > available(n, vector<bool>(n,true));
    solve(available,k);
    
    cout << ans/factorial(k) << endl;
    
}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 1

Verdict: ACCEPTED

Test 3

Group: 1

Verdict: ACCEPTED

Test 4

Group: 1

Verdict: TIME LIMIT EXCEEDED

Test 5

Group: 1

Verdict: TIME LIMIT EXCEEDED

Test 6

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 7

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 9

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 10

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 11

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 12

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 13

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 14

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 15

Group: 3

Verdict: TIME LIMIT EXCEEDED