CSES - Datatähti 2025 alku - Results
Submission details
Task:Kortit II
Sender:Vludo
Submission time:2024-11-03 18:52:22 +0200
Language:C++ (C++20)
Status:READY
Result:8
Feedback
groupverdictscore
#1ACCEPTED3
#2ACCEPTED5
#30
#40
#50
Test results
testverdicttimegroup
#1ACCEPTED0.00 s1, 2, 3, 4, 5details
#2ACCEPTED0.00 s2, 3, 4, 5details
#30.01 s3, 4, 5details
#40.01 s4, 5details
#50.00 s5details
#60.00 s5details

Code

#include<bits/stdc++.h>
using namespace std;
#define forn(i,n) for(int i=0;i<n;++i)
#define forN(i,k,n) for(int i=k;i<n;++i)
#define pb push_back
#define F first
#define S second
typedef long long ll;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
#define inf ((int)(1.01e9))
#define mod ((int)(1e9+7))
#define N ((int)(102))
 
ll n, m, a, b, counter, result, fact[N], A[N][N];
unordered_map<ll,ll> mem;
vector<int> v1, v2;

ll nCr(int n, int r);
ll dcmb(int n, int a) {
    ll counter = 0; 
    forn(i, n+1) 
        if (n-i >= a) {
            ll t = (nCr(n,i) * A[n-i][a]) % mod;
            if (i&1) t = (mod-t)%mod; 
            counter = (counter+t) % mod;
        }
    return counter;
}
void solve();
 
int main() {
    cin.tie(0) -> sync_with_stdio(0);
    cout.tie(0);
    fact[0] = fact[1] = 1;
    forN(i,2,N) fact[i] = ((i)*fact[i-1]) %mod;

    A[0][0] = 1;
    forN(i, 1, N) {
        A[i][0] = A[i-1][0];
        forN(k, 1, i) {
            A[i][k] = ((k+1)*A[i-1][k] + (i-k)*A[i-1][k-1])%mod;
        }
    }
 
    int t;
    cin >> t;
    while(t--) solve();
}

ll nCr(int n, int r) {
    if (!r) return 1;
    return (fact[n] / ((fact[r] * fact[n-r])%mod)) % mod;
}

void solve() {
    cin >> m >> a >> b;
    if (m < a + b || ((!b)&&a) || ((!a)&&b)) {
        cout << "0" << endl;
        return;
    }
    n = a + b;
 
    ll comb = nCr(m,m-n);
    comb = (comb*comb*fact[m-n]*fact[n])%mod;

    result = (dcmb(n, a)*comb) % mod;
    cout << result << endl;
}

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:

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

correct output
40291066
0
0
0
0
...

user output
126727510
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
255738551
...

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

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)