Lähetit

#include <iostream>
#include <bitset>
 
using namespace std;
 
typedef long long int ll;
 
template<ll N>
struct modnum {
    ll num;
    modnum(ll n) :num(n%N) {}
 
    modnum &operator+=(const modnum<N> &o) {
        num = (num + o.num) % N;
        return *this;
    }
    modnum &operator*=(const modnum<N> &o) {
        num = (num*o.num) % N;
        return *this;
    }
    modnum operator*(const modnum<N> &o) const {
        modnum<N> a = *this;
        a *= o;
        return a;
    }
 
    operator ll() const {
        return num;
    }
};
typedef modnum<1000000007> num;
//typedef ll num;
 
typedef bitset<100> BS;
 
num testlayer(int n, int k, int l, BS b) {
    if (k <= 0) {
        return 1;
    }
    if (l >= n) {
        return 0;
    }
    
    num r = testlayer(n, k, l + 1, b);
 
    for (int i = 0; i < 
         l / 2 * 2
         + (n + 1) % 2
         +1;
        i++) {
        if (!b.test(i)) {
            b.set(i);
            r += testlayer(n, k - 1, l + 1, b);
            b.reset(i);
        }
    }
 
    return r;
}
 
num testdiag(int n, int k) {
    BS b;
    return testlayer(n, k, 0, b);
}
 
num test(int n, int k) {
    num r = 0;
 
    for (int i = 0; i <= k; i++) {
        r += testdiag(n, i)*testdiag(n-1, k - i);
    }
 
    return r;
}
 
int main() {
    /*cout << "\t";
    for (int x = 1; x < 10; x++)
        cout << x << "\t";
    for (int y = 0; y < 10; y++) {
        cout << y << "|\t";
        for (int x = 1; x < 10; x++) {
            cout << testdiag(x, y) << "\t";
        }
        cout << "\n";
    }*/
    int n, k;
    cin >> n >> k;
    cout << test(n, k);
}

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: WRONG ANSWER

Test 7

Group: 2

Verdict: WRONG ANSWER

Test 8

Group: 2

Verdict: WRONG ANSWER

Test 9

Group: 2

Verdict: WRONG ANSWER

Test 10

Group: 2

Verdict: ACCEPTED

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