CSES - Datatähti 2025 alku - Results
Submission details
Task:Kortit II
Sender:aatukaj
Submission time:2024-10-28 23:27:52 +0200
Language:C++ (C++20)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
#40
#50
Test results
testverdicttimegroup
#1--1, 2, 3, 4, 5details
#2--2, 3, 4, 5details
#3--3, 4, 5details
#4--4, 5details
#5--5details
#6--5details

Code

// #pragma GCC optimize("O3,unroll-loops")
// #pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")

#include <bits/stdc++.h>
#include<ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>

using namespace std;
using namespace __gnu_pbds;

template<class T> using indexed_set = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;

using ll = long long;
using ull = unsigned long long;
using ld = long double;

using pii = pair<int, int>;
using pll = pair<ll, ll>;

// const ll M = 998244353;
const ll M = 1e9 + 7;
// ll M;


#define all(v) (v).begin(), (v).end()
#define ff first
#define ss second
#define PI 3.141592653589793238462643383279502884L

const ll INFL = 1e18;
const int INF = 1e9;
const ll maxN = 5e5+1;

ll inv(ll a) {
    return a<=1 ? a: M - (ll)(M/a) * inv(M%a) % M;
}

ll modpow(ll a, ll b) {
    a %= M;
    ll res = 1;
    while (b > 0) {
        if (b & 1)
            res = res * a % M;
        a = a * a % M;
        b >>= 1;
    }
    return res;
}

ll fact[maxN];
ll inv_fact[maxN];
ll t[100][100];

void calc_fact() {
    fact[0] = 1;
    for (int i=1; i<maxN; i++) {
        fact[i] = fact[i-1]*i % M;
        inv_fact[i] = inv(fact[i]);
    }
}

ll nCr(int n, int k) {
    return fact[n] * inv(fact[k]*fact[n-k] % M) % M;
}


ll dp2[2001][2001];

ll get(ll n, ll a, ll b) {
    if (a==0 && b==0) return 1;
    if (a+b>n) return 0;
    // return dp[n][a][b];
    return nCr(n, n-(a+b))*dp2[a+b][a]%M;
}

void solve() {
    int n, a, b;
    cin >> n >> a >> b;
    cout << get(n, a, b)*fact[n]%M << '\n';
    // cout << ans*fact[n]%M << '\n'; 

    
}
int main() {
    calc_fact();
    ios_base::sync_with_stdio(0); cin.tie(NULL);
    // cout << setprecision(12);
    // for (ll n=1; n<=100; n++) {
    //     dp[n][0][0] = 1;
    //     for (ll a=1; a<=n; a++) {
    //         for (ll b=1; b<=n; b++) {
    //             if (a+b>n) continue;
    //             dp[n][a][b] = dp[n-1][a][b];
    //             dp[n][a][b] += (n-a-b+1)*(dp[n-1][a-1][b-1]);
    //             dp[n][a][b] += a*dp[n-1][a][b-1];
    //             dp[n][a][b] += b*dp[n-1][a-1][b];
    //             dp[n][a][b] %= M;
    //             assert(nCr(n, n-(a+b))*dp[a+b][a][b]%M==dp[n][a][b]);
    //         }
    //     }
    //
    // }
    for (ll n=1; n<=2000; n++) {
        for (ll a=1; a<n; a++) {
            ll b = n-a;
            dp2[n][a] = get(n-1,a,b);
            dp2[n][a] += (n-a-b+1)*get(n-1,a-1, b-1);
            dp2[n][a] += a*get(n-1,a,b-1);
            dp2[n][a] += b*get(n-1,a-1,b);
            dp2[n][a] %= M;
            // cout << n << ' ' << a << ' ' << b << endl;
            // cout << dp2[n][a] << ' ' << dp[n][a][b] << endl;
            // cout << endl;
            // assert(dp2[n][a]==dp[n][a][b]);
        }

    }


    int tc = 1;
    cin >>tc;
    while (tc--) solve();

}

Test details

Test 1

Group: 1, 2, 3, 4, 5

Verdict:

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

correct output
0
0
0
0
0
...

user output
(empty)

Test 2

Group: 2, 3, 4, 5

Verdict:

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

correct output
0
0
35280
0
36720
...

user output
(empty)

Test 3

Group: 3, 4, 5

Verdict:

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

correct output
40291066
0
0
0
0
...

user output
(empty)

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
(empty)

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
(empty)

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
(empty)