Kortit II

#include<bits/stdc++.h>
using namespace std;
 
#define debug(x) cout << #x << " = " << x << "\n"
#define pb push_back
 
typedef long long ll;
typedef vector<ll> vi;
typedef pair<ll,ll> pi;
 
const ll M = 1000000007;
const ll INF = LLONG_MAX/2-1;
 
ll g[2004][2004]; 
void triangle(){
    g[2][1] = 1;
    for(int i=2; i<=2001; i++){
        for(int r=0; r<=i; r++){
            g[i+1][r] = (r*g[i][r] + (i+1-r)*g[i][r-1] + i*g[i-1][r-1])%M; 
        }
    }
}

vi fact(2002);
void factorials(){
    int sol=1;
    for(int i=2; i<2002; i++){
        fact[i-1] = sol;
        sol = (sol*i)%M;
    }
}

ll nCr(ll in, ll ir){
    if(ir==0 || ir==in){
        return 1;
    }else{
        return nCr(in-1,ir-1)%M + nCr(in-1,ir)%M;
    }
}

void solve(){
    int n,a,b; cin >> n >> a >> b;
    if(a+b>n || ((a==0)^(b==0))){
        cout << 0 << "\n";
    }else if(a==0 && b==0){
        cout << fact[n] << "\n";
    }else{
        cout << (nCr(n,n-a-b)%M * g[a+b][a]%M)%M * fact[n]%M << "\n";
    }
}
 
int main(){
    ios::sync_with_stdio(0);
    cin.tie(0);
 
    int t=1; cin >> t;
    triangle();
    factorials();

    while(t--) solve();
 
    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: WRONG ANSWER

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