Kortit II

input/code.cpp:22:19: warning: use of 'auto' in parameter declaration only available with '-std=c++20' or '-fconcepts'
   22 | void print(vector<auto> v) {
      |                   ^~~~

#include <bits/stdc++.h>

#define F0R(n) for (int i=0; i<n; i++)
#define FOR(from, to) for (int i = from; i < to; i++)
#define R0F(n) for (int i = n; i>=0; --i)
#define ROF(from, to) for (int i = from; i >= to; --i)

#define ll long long
#define mod (int) (1e9 + 7)
#define vi vector<int>
#define vll vector<ll>
#define vc vector<char>
#define pii pair<int, int>
#define pci pair<char, int>
#define pll pair<ll, ll>
#define YES cout << "YES\n";
#define NO cout << "NO\n";
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
using namespace std;

void print(vector<auto> v) {
	for (auto it = v.begin(); it != v.end(); ++it)
		cout << *it << " ";
	cout << " ";
}

vector<vi> adj(int(1e5+1), vi());
vi vis(int(1e5+1));

bool valid(vi& p2, int a, int b) {
    int sca=0, scb=0, n = p2.size();
    F0R (n) {
        if (p2[i]<i+1)
            ++sca;
        else if (p2[i]>i+1)
            ++scb;
        if (sca > a || scb > b)
            return false;
    }
    if (sca == a && scb == b)
        return true;
    else
        return false;
}

pii score(vi& p2) {
    int sca=0, scb=0, n = p2.size();
    F0R (n) {
        if (p2[i]<i+1)
            ++sca;
        else if (p2[i]>i+1)
            ++scb;
    }
    return {sca, scb};
}

void soln() {
    int n, a, b;
    cin >> n >> a >> b;

    ll ans = 0, fact_mod = 1;

    F0R (n) {
        (fact_mod *= i+1) %= mod;
    }
    // printf("fact_mod=%lld", fact_mod);

    set<int> st;
    // vi p1(n);
    vi p2(n, 0);
    iota(p2.begin(), p2.end(), 1);
    // F0R (n) p1[i]=i+1, p2[i] = i+1;

    if (valid(p2, a, b)) {
        ans += fact_mod;
        ans %= mod;
    }

    // print(p1);
    while (next_permutation(p2.begin(), p2.end())) {
        if (valid(p2, a, b)) {
            // print(p2);
            // auto [sca, scb]
            ans += fact_mod;
            ans %= mod;
        }
        else {
            // print(p2);
            auto [sca, scb] = score(p2);
            if (sca > a && scb > b)
                break;
            // cout << sca << "  " << scb << endl;
        }
    }
    cout << ans << "\n";
}

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

    int tc;
    cin >> tc;
    while (tc--) soln();
    return 0;
}

Test 1

Group: 1, 2, 3, 4, 5

Verdict: WRONG ANSWER

Test 2

Group: 2, 3, 4, 5

Verdict: WRONG ANSWER

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