QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#340178 | #1810. Generate the Sequences | HKOI0# | AC ✓ | 55ms | 104912kb | C++20 | 1.3kb | 2024-02-28 17:35:53 | 2024-02-28 17:35:54 |
Judging History
answer
#include <bits/stdc++.h>
#define int long long
using namespace std;
using ll = long long;
const int N = 2e6 + 11;
const int MOD = 998244353;
int fa[N], fi[N];
int pm(int a, int b){
if (b == 0) return 1;
return pm(a * a % MOD, b / 2) * (b % 2 ? a : 1) % MOD;
}
int mi(int a){
return pm(a, MOD - 2);
}
int binom(int n, int r){
return fa[n] * fi[r] % MOD * fi[n - r] % MOD;
}
template<int MOD = 998244353>
struct Mint{
};
void solve() {
int n,m;
cin>>n>>m;
vector dp(n+2,vector<ll>(n+2));
dp[0][0]=1;
for(int i=0;i<n;i++)
for(int j=0;j<=i;j++) {
dp[i+1][j+1]=(dp[i+1][j+1]+dp[i][j])%MOD;
if(i-j<(ll)j*(m-2)) dp[i+1][j]=(dp[i+1][j]+dp[i][j]*(((ll)j*(m-2)%MOD-(i-j)+MOD)%MOD)%MOD)%MOD;
}
ll ans=0;
for(int i=0;i<=n;i++) {
ll sum=0;
for(int j=0;j<=i;j++) sum=(sum+dp[i][j])%MOD;
ans=(ans+sum*binom(n,i))%MOD;
}
cout<<ans<<'\n';
}
int32_t main() {
#ifndef LOCAL
ios::sync_with_stdio(false);
cin.tie(nullptr);
fa[0] = 1; for (int i = 1; i < N; i++) fa[i] = fa[i - 1] * i % MOD;
fi[N - 1] = mi(fa[N - 1]); for (int i = N - 2; i >= 0; i--) fi[i] = fi[i + 1] * (i + 1) % MOD;
#endif
int T = 1;
// cin >> T;
while (T--) solve();
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 12ms
memory: 34780kb
input:
2 3
output:
5
result:
ok answer is '5'
Test #2:
score: 0
Accepted
time: 12ms
memory: 42860kb
input:
1024 52689658
output:
654836147
result:
ok answer is '654836147'
Test #3:
score: 0
Accepted
time: 13ms
memory: 34880kb
input:
1 2
output:
2
result:
ok answer is '2'
Test #4:
score: 0
Accepted
time: 12ms
memory: 34780kb
input:
1 3
output:
2
result:
ok answer is '2'
Test #5:
score: 0
Accepted
time: 20ms
memory: 35064kb
input:
1 100000000
output:
2
result:
ok answer is '2'
Test #6:
score: 0
Accepted
time: 8ms
memory: 34784kb
input:
2 2
output:
4
result:
ok answer is '4'
Test #7:
score: 0
Accepted
time: 20ms
memory: 34800kb
input:
2 4
output:
6
result:
ok answer is '6'
Test #8:
score: 0
Accepted
time: 18ms
memory: 34836kb
input:
2 5
output:
7
result:
ok answer is '7'
Test #9:
score: 0
Accepted
time: 16ms
memory: 34732kb
input:
2 100000000
output:
100000002
result:
ok answer is '100000002'
Test #10:
score: 0
Accepted
time: 15ms
memory: 34780kb
input:
3 2
output:
8
result:
ok answer is '8'
Test #11:
score: 0
Accepted
time: 11ms
memory: 34784kb
input:
3 3
output:
14
result:
ok answer is '14'
Test #12:
score: 0
Accepted
time: 19ms
memory: 34732kb
input:
3 4
output:
22
result:
ok answer is '22'
Test #13:
score: 0
Accepted
time: 8ms
memory: 34776kb
input:
3 5
output:
32
result:
ok answer is '32'
Test #14:
score: 0
Accepted
time: 15ms
memory: 34880kb
input:
3 100000000
output:
446563791
result:
ok answer is '446563791'
Test #15:
score: 0
Accepted
time: 46ms
memory: 104828kb
input:
3000 2
output:
21292722
result:
ok answer is '21292722'
Test #16:
score: 0
Accepted
time: 43ms
memory: 104776kb
input:
3000 3
output:
172222927
result:
ok answer is '172222927'
Test #17:
score: 0
Accepted
time: 48ms
memory: 104912kb
input:
3000 100000000
output:
736503947
result:
ok answer is '736503947'
Test #18:
score: 0
Accepted
time: 53ms
memory: 84224kb
input:
2522 61077387
output:
857454425
result:
ok answer is '857454425'
Test #19:
score: 0
Accepted
time: 11ms
memory: 35900kb
input:
426 7215704
output:
799491736
result:
ok answer is '799491736'
Test #20:
score: 0
Accepted
time: 20ms
memory: 39036kb
input:
772 72289915
output:
848141383
result:
ok answer is '848141383'
Test #21:
score: 0
Accepted
time: 23ms
memory: 51000kb
input:
1447 83321470
output:
160422285
result:
ok answer is '160422285'
Test #22:
score: 0
Accepted
time: 35ms
memory: 83312kb
input:
2497 64405193
output:
355300540
result:
ok answer is '355300540'
Test #23:
score: 0
Accepted
time: 10ms
memory: 39036kb
input:
775 9385367
output:
470172346
result:
ok answer is '470172346'
Test #24:
score: 0
Accepted
time: 15ms
memory: 42016kb
input:
982 72596758
output:
7144187
result:
ok answer is '7144187'
Test #25:
score: 0
Accepted
time: 16ms
memory: 35656kb
input:
417 26177178
output:
776374896
result:
ok answer is '776374896'
Test #26:
score: 0
Accepted
time: 32ms
memory: 63596kb
input:
1932 19858856
output:
285834553
result:
ok answer is '285834553'
Test #27:
score: 0
Accepted
time: 55ms
memory: 92568kb
input:
2728 23009122
output:
433516287
result:
ok answer is '433516287'
Test #28:
score: 0
Accepted
time: 30ms
memory: 61616kb
input:
1857 22578508
output:
243488639
result:
ok answer is '243488639'
Test #29:
score: 0
Accepted
time: 52ms
memory: 101100kb
input:
2918 69623276
output:
546299707
result:
ok answer is '546299707'
Test #30:
score: 0
Accepted
time: 34ms
memory: 56424kb
input:
1679 21332149
output:
217000656
result:
ok answer is '217000656'
Test #31:
score: 0
Accepted
time: 17ms
memory: 48520kb
input:
1340 6251797
output:
267221018
result:
ok answer is '267221018'
Test #32:
score: 0
Accepted
time: 26ms
memory: 40300kb
input:
868 64770398
output:
652067665
result:
ok answer is '652067665'