CSES - Datatähti 2025 alku - Results
Submission details
Task:Kortit II
Sender:aatukaj
Submission time:2024-10-28 18:41:27 +0200
Language:C++ (C++20)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
#40
#50
Test results
testverdicttimegroup
#10.24 s1, 2, 3, 4, 5details
#20.24 s2, 3, 4, 5details
#30.24 s3, 4, 5details
#40.24 s4, 5details
#50.24 s5details
#60.24 s5details

Code

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

#include <algorithm>
#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;
}

void solve() {
    int n, a, b;
    cin >> n >> a >> b;
    if (a+b>n) {
        cout << "0\n";
        return;
    }
    if ((a && !b) || (b && !a)) {
        cout << "0\n";
        return;
    }
    ll ans = fact[n]*nCr(n, n-a-b)%M;
    ans = ans * t[a+b][a] % M;
    cout << ans << '\n';
    
}
int main() {
    calc_fact();
    ios_base::sync_with_stdio(0); cin.tie(NULL);
    // cout << setprecision(12);
    assert(false);
    for (int sz=1; sz<=8; sz++) {
        for (int i=1; i<=sz; i++) {
            vector<int> p(sz);
            iota(all(p), 0);
            int cur = 0;
            do { 
                int a = 0;
                for (int j=0; j<sz; j++) {
                    if (p[j]==j) {
                        a = -INF;
                        break;
                    }
                    a += p[j]>j;
                }
                cur += a==i;

            } while (next_permutation(all(p)));
            t[sz][i] = cur;
            // cout << cur << ' ';
        }
        // cout << '\n';

    }


    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)

Error:
code: input/code.cpp:87: int main(): Assertion `false' failed.

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)

Error:
code: input/code.cpp:87: int main(): Assertion `false' failed.

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)

Error:
code: input/code.cpp:87: int main(): Assertion `false' failed.

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)

Error:
code: input/code.cpp:87: int main(): Assertion `false' failed.

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)

Error:
code: input/code.cpp:87: int main(): Assertion `false' failed.

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)

Error:
code: input/code.cpp:87: int main(): Assertion `false' failed.