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