CSES - Datatähti 2025 alku - Results
Submission details
Task:Kortit II
Sender:PMak
Submission time:2024-11-09 00:37:00 +0200
Language:C++ (C++17)
Status:READY
Result:34
Feedback
groupverdictscore
#1ACCEPTED3
#2ACCEPTED5
#3ACCEPTED26
#40
#50
Test results
testverdicttimegroup
#1ACCEPTED0.15 s1, 2, 3, 4, 5details
#2ACCEPTED0.15 s2, 3, 4, 5details
#3ACCEPTED0.15 s3, 4, 5details
#40.15 s4, 5details
#50.15 s5details
#60.18 s5details

Code

#include <bits/stdc++.h>

#define F0R(n) for (int i=0; i<n; i++)
#define FOR(i, from, to) for (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 (ll) (1e9 + 7)
#define vll vector<ll>
using namespace std;

ll N = 2501;

vll fact_mod(2*N+2, 0);
void build_fact() {
    fact_mod[0] = 1ll;
    ll i;
    FOR (i, 1ll, 2*N+2)
        // fact_mod[i] = fact_mod[i-1]*i;
        (fact_mod[i] = fact_mod[i-1]*i) %= mod;
}

ll nCr(ll n, ll k) {
    if (k <= 0) return 1ll;
    ll ans = 1ll, i;
    FOR (i, n-k+1, n+1)
        ans *= i;
    FOR (i, 1, k+1)
        ans /= i;

    ans %= mod;
    return ans;
}

vector<vll> dp(N, vll(N, 0));
void build() {
    dp[0][0] = 1ll;
    ll a, b;
    FOR (a, 1, N) {
        FOR (b, 1, N) {
            (dp[a][b] = dp[a][b-1]*a) %= mod;
            (dp[a][b] += dp[a-1][b]*b) %= mod;
            (dp[a][b] += dp[a-1][b-1]*(a+b-1ll)) %= mod;
            // (dp[a][b] *= fact_mod[a+b]) %= mod;
        }
    }
}

void soln() {
    ll n, a, b, ans;
    cin >> n >> a >> b;
    if (a+b>n) {
        ans = 0;
        cout << ans << "\n";
        return;
    }

    ans = dp[a][b];
    ans *= nCr(n, a+b);
    ans %= mod;
    ans *= fact_mod[n];
    ans %= mod;
    cout << ans << "\n";
}

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

    build_fact();
    build();

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

Test details

Test 1

Group: 1, 2, 3, 4, 5

Verdict: ACCEPTED

input
54
4 4 0
3 1 3
3 2 2
4 0 4
...

correct output
0
0
0
0
0
...

user output
0
0
0
0
0
...

Test 2

Group: 2, 3, 4, 5

Verdict: ACCEPTED

input
284
6 1 0
5 0 2
7 1 5
7 7 5
...

correct output
0
0
35280
0
36720
...

user output
0
0
35280
0
36720
...

Test 3

Group: 3, 4, 5

Verdict: ACCEPTED

input
841
19 3 12
19 19 13
19 7 13
20 11 15
...

correct output
40291066
0
0
0
0
...

user output
40291066
0
0
0
0
...

Test 4

Group: 4, 5

Verdict:

input
1000
15 12 6
7 1 6
44 4 26
6 6 5
...

correct output
0
5040
494558320
0
340694548
...

user output
0
5040
0
0
0
...

Test 5

Group: 5

Verdict:

input
1000
892 638 599
966 429 655
1353 576 1140
1403 381 910
...

correct output
0
0
0
249098285
0
...

user output
0
0
0
0
0
...

Test 6

Group: 5

Verdict:

input
1000
2000 1107 508
2000 1372 249
2000 588 65
2000 1739 78
...

correct output
750840601
678722180
744501884
159164549
868115056
...

user output
0
0
0
0
0
...