QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#364711 | #3161. Another Coin Weighing Puzzle | chuchu# | TL | 871ms | 4616kb | C++23 | 795b | 2024-03-24 16:11:51 | 2024-03-24 16:11:53 |
Judging History
answer
#include<bits/stdc++.h>
using namespace std;
using ll = long long;
constexpr ll P = 998244353;
ll qpow(ll a, ll b) {
ll ans = 1;
for(; b; b/=2, a = a * a % P) {
if(b%2==1) ans = ans * a % P;
}
return ans;
}
void solve() {
ll m, k; cin >> m >> k;
vector<ll>dp(k+1);
for(int i = 1 ; i <= k ; i ++) {
dp[i] = qpow(2*i+1, m);
}
for(int i = 1 ; i <= k ; i ++) {
for(int d = 2 ; d <= i ; ) {
int k = i / d;
int nxt = min(i, i/k);
dp[i] = (dp[i] - (dp[k] - 1) * (nxt-d+1)) % P;
d = nxt+1;
}
}
cout << (dp[k] + P) % P << endl;
}
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
solve();
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 3520kb
input:
2 1
output:
9
result:
ok single line: '9'
Test #2:
score: 0
Accepted
time: 1ms
memory: 3488kb
input:
2 2
output:
17
result:
ok single line: '17'
Test #3:
score: 0
Accepted
time: 10ms
memory: 3564kb
input:
10000 10000
output:
689223145
result:
ok single line: '689223145'
Test #4:
score: 0
Accepted
time: 1ms
memory: 3584kb
input:
9999 31
output:
986106162
result:
ok single line: '986106162'
Test #5:
score: 0
Accepted
time: 10ms
memory: 3636kb
input:
57 9817
output:
447253096
result:
ok single line: '447253096'
Test #6:
score: 0
Accepted
time: 0ms
memory: 3572kb
input:
501 499
output:
247755220
result:
ok single line: '247755220'
Test #7:
score: 0
Accepted
time: 719ms
memory: 4316kb
input:
97424 174829
output:
964884269
result:
ok single line: '964884269'
Test #8:
score: 0
Accepted
time: 0ms
memory: 3560kb
input:
11 13
output:
729153057
result:
ok single line: '729153057'
Test #9:
score: 0
Accepted
time: 871ms
memory: 4616kb
input:
200000 200000
output:
803771125
result:
ok single line: '803771125'
Test #10:
score: 0
Accepted
time: 1ms
memory: 3816kb
input:
199999 562
output:
865836540
result:
ok single line: '865836540'
Test #11:
score: 0
Accepted
time: 806ms
memory: 4604kb
input:
3539 189423
output:
530738158
result:
ok single line: '530738158'
Test #12:
score: 0
Accepted
time: 712ms
memory: 4368kb
input:
198324 173852
output:
963717515
result:
ok single line: '963717515'
Test #13:
score: 0
Accepted
time: 0ms
memory: 3824kb
input:
1 1
output:
3
result:
ok single line: '3'
Test #14:
score: -100
Time Limit Exceeded
input:
1000000 1000000