Task: | Kortit II |
Sender: | aatukaj |
Submission time: | 2024-10-28 23:31:05 +0200 |
Language: | C++ (C++20) |
Status: | READY |
Result: | 100 |
group | verdict | score |
---|---|---|
#1 | ACCEPTED | 3 |
#2 | ACCEPTED | 5 |
#3 | ACCEPTED | 26 |
#4 | ACCEPTED | 28 |
#5 | ACCEPTED | 38 |
test | verdict | time | group | |
---|---|---|---|---|
#1 | ACCEPTED | 0.29 s | 1, 2, 3, 4, 5 | details |
#2 | ACCEPTED | 0.29 s | 2, 3, 4, 5 | details |
#3 | ACCEPTED | 0.29 s | 3, 4, 5 | details |
#4 | ACCEPTED | 0.29 s | 4, 5 | details |
#5 | ACCEPTED | 0.29 s | 5 | details |
#6 | ACCEPTED | 0.29 s | 5 | details |
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 ... |