Kortit II

#include <iostream>
#include <vector>
using namespace std;


/*
Nykynen suunnitelma:
1. Peli voi kulkea n! eri tavalla, sillä samat kortit pelattuna pareittain eri järjestyksessä johtaa samaan tulokseen
2. Tämän jälkeen katsotaan jokaiselle numerolle, mikä on pienin ja suurin luku millä sen voi korvata
3. Toistetaan askel 2 memoisaatiolla, kunnes ollaan valmiita
*/

int n;
int mod = 1e9+7;

struct cell {
    int first;
    int second;
};


int dfs ( vector<int>& used, cell score, const cell& target, int index ) {
    if ( index == n) {
        if ( score.first == target.first && score.second == target.second )
            return 1;
        else
            return 0;
    }

    int counter = 0;

    for ( int i = 1; i <= n; i++ ) {
        if ( used[i] == 2 )
            continue;
        
        used[i]++;

        cell next_score = score;
        next_score.first += (index+1) > i;
        next_score.second += (index+1) < i;

        counter += dfs(used, next_score, target, index+1);
        counter %= mod;

        used[i]--;
    }

    return counter;
}


int main () {
    int t;
    cin >> t;

    int ans[t];
    
    for ( int i = 0; i < t; i++ ) {
        
        int a, b;

        cin >> n >> a >> b;

        if ( (a+b) > n || (a == 0 && b != 0) || (a != 0 && b == 0)) {
            ans[i] = 0;
            continue;
        }

        cell score = {0, 0};
        cell target = {a, b};
        vector<int> used(n+1, 1);
        
        ans[i] = dfs(used, score, target, 0);

        for (int j = 2; j <= n; j++) {
            ans[i] *= j;
            ans[i] %= mod;
        }
    }

    for (auto n : ans)
        cout << n << "\n";
}

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