Kortit II

#include <algorithm>
#include <iterator>
#include <vector>
#include <iostream>
 
const long MOD = 1000000007;
 
long factorial(long n) {
    long ans = 1;
    for (long i = 1; i <= n; ++i) {
        ans = (ans * i) % MOD;
    }
    return ans;
}
 
struct game {
    long n;
    long a;
    long b;
};
 
long recurse(std::vector<long>& px, long n, long lt, long gt, long max) {
    long sum = 0;
 
    std::vector<long> nums{};
 
    for (long i = 0; i < max; ++i) {
        nums.push_back(i);
    }
 
    if (n >= max) {
        return 1;
    }
 
    if (lt > 0) {
        for (long i = 0; i < n; ++i) {
            if (std::find(px.begin(), px.end(), i) == std::end(px)) {
                px.push_back(i);
                sum += recurse(px, n+1, lt-1, gt, max);
                px.pop_back();
            } else {
            }
        }
    }
 
    if (gt > 0) {
        for (long i = n+1; i < max; ++i) {
            if (std::find(px.begin(), px.end(), i) == std::end(px)) {
                px.push_back(i);
                sum += recurse(px, n+1, lt, gt-1, max);
                px.pop_back();
            }
        }
    }
 
    return sum;
}
 
int main() {
    long count;
    std::cin >> count;
    std::vector<game> games{};
 
    for (long i = 0; i < count; ++i) {
        long n, a, b;
 
        std::cin >> n >> a >> b;
        games.push_back(game{ .n = n, .a = a, .b = b });
    }
 
    for (auto [n, a, b] : games) {
        if (a + b > n || ((a != b) && (a == 0 || b == 0))) {
            std::cout << 0 << "\n";
            continue;
        }
 
        long total = ((factorial(n) / factorial(n-(a+b))) * (factorial(n) / factorial(a+b))) % MOD;
 
        std::vector<long> px{};
 
        long sum = recurse(px, 0, a, b, a+b);
 
        std::cout << (sum * total) % MOD << "\n";
    }
    return 0;
}

Test 1

Group: 1, 2, 3, 4, 5

Verdict: ACCEPTED

Test 2

Group: 2, 3, 4, 5

Verdict: ACCEPTED

Test 3

Group: 3, 4, 5

Verdict: TIME LIMIT EXCEEDED

Test 4

Group: 4, 5

Verdict: TIME LIMIT EXCEEDED

Test 5

Group: 5

Verdict: TIME LIMIT EXCEEDED

Test 6

Group: 5

Verdict: TIME LIMIT EXCEEDED