CSES - Datatähti 2025 alku - Results
Submission details
Task:Kortit II
Sender:aatukaj
Submission time:2024-10-28 23:31:05 +0200
Language:C++ (C++20)
Status:READY
Result:100
Feedback
groupverdictscore
#1ACCEPTED3
#2ACCEPTED5
#3ACCEPTED26
#4ACCEPTED28
#5ACCEPTED38
Test results
testverdicttimegroup
#1ACCEPTED0.29 s1, 2, 3, 4, 5details
#2ACCEPTED0.29 s2, 3, 4, 5details
#3ACCEPTED0.29 s3, 4, 5details
#4ACCEPTED0.29 s4, 5details
#5ACCEPTED0.29 s5details
#6ACCEPTED0.29 s5details

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 C[2001][2001];
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 C[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]);
    //         }
    //     }
    //
    // }
    C[0][0] = 1;
    for (ll n=1; n<=2000; n++) {
        C[n][0] = C[n][n] = 1;
        for (ll k=1; k<n; k++) {
            C[n][k] = (C[n-1][k-1]+C[n-1][k])%M;
        }

    }

    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: 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: ACCEPTED

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
494558320
0
340694548
...

Test 5

Group: 5

Verdict: ACCEPTED

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
249098285
0
...

Test 6

Group: 5

Verdict: ACCEPTED

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

correct output
750840601
678722180
744501884
159164549
868115056
...

user output
750840601
678722180
744501884
159164549
868115056
...