Lähetit

#include <iostream>
#include <vector>
#include <algorithm>
#include <cstdint>
#include <numeric>

using namespace std;

typedef int64_t ll;

void print(vector<bool>& a, int n)
{
    cout << endl;
    for (int j = 0; j < n; ++j) {
        for (int k = 0; k < n; ++k) {
            int l = j*n+k;
            cout << (a[l] ? '1' : '0') << ' ';
        }
        cout << endl;
    }
}

void set(vector<bool>& a, int n, int i)
{
    int c = i%n;
    for (int j = i, k = c; j >= 0  && k >= 0; j -= n+1, --k) a[j] = true;
    for (int j = i, k = c; j >= 0  && k <  n; j -= n-1, ++k) a[j] = true;
    for (int j = i, k = c; j < n*n && k >= 0; j += n-1, --k) a[j] = true;
    for (int j = i, k = c; j < n*n && k <  n; j += n+1, ++k) a[j] = true;
}

const ll m = 1000000007;

ll search(const vector<bool>& a, int n, int k, int j)
{
    if (k == 0) return 1;
    ll s = 0;
    for (int i = j; i < n*n; ++i) {
        if (a[i]) continue;
        vector<bool> b(a);
        set(b, n, i);
        s += search(b, n, k-1, i);
        s %= m;
    }
    return s;
}

int main()
{
    int n, k;
    cin >> n >> k;
    vector<bool> a(n*n);
    cout << search(a,n,k,0) << endl;
    return 0;
}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 1

Verdict: ACCEPTED

Test 3

Group: 1

Verdict: ACCEPTED

Test 4

Group: 1

Verdict: ACCEPTED

Test 5

Group: 1

Verdict: ACCEPTED

Test 6

Group: 2

Verdict: ACCEPTED

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