QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#364443 | #3161. Another Coin Weighing Puzzle | chuchu# | WA | 5ms | 23528kb | C++14 | 1.4kb | 2024-03-24 14:31:58 | 2024-03-24 14:31:58 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
#define debug(x) cerr << #x << " == " << x << endl
#define el '\n'
#define fir first
#define sec second
typedef long long ll;
typedef pair<int, int> PII;
const int mod = 998244353;
const int inf = 0x3f3f3f3f;
const int N = 1000020;
int phi[N], vis[N], prime[N];
ll sum[N];
ll up(ll x)
{
return (x % mod + mod) % mod;
}
void solve()
{
int n, k;
cin >> n >> k;
vector<ll> ans(n + 1, 0), fac(n + 1, 1);
ll cnt = sum[k];
for (int i = 1; i <= n; i++) {
fac[i] = up(fac[i - 1] * (2 * k + 1));
}
// cout << cnt << endl;
ans[1] = 3;
for (int m = 2; m <= n; m++) {
ans[m] = ans[m - 1] + 2 * (2 * cnt + 1) % mod * fac[m - 2] % mod;
ans[m] = up(ans[m]);
}
cout << ans[n] << el;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);
int T = 1;
// cin >> T;
phi[1] = 1;
int cnt = 0;
for (int i = 2; i <= 1000000; i++) {
if (!vis[i]) {
vis[i] = i;
prime[cnt++] = i;
phi[i] = i - 1;
}
for (int j = 0; j < cnt; j++) {
if (i * prime[j] > 1000000)
break;
vis[i * prime[j]] == prime[j];
if (i % prime[j] == 0) {
phi[i * prime[j]] = phi[i] * prime[j];
break;
}
phi[i * prime[j]] = phi[i] * phi[prime[j]];
}
}
sum[1] = 1;
for (int i = 2; i <= 1000000; i++) {
sum[i] = sum[i - 1] + phi[i] * 2;
}
while (T--) {
solve();
}
return 0;
}
详细
Test #1:
score: 100
Accepted
time: 5ms
memory: 23048kb
input:
2 1
output:
9
result:
ok single line: '9'
Test #2:
score: 0
Accepted
time: 3ms
memory: 23152kb
input:
2 2
output:
17
result:
ok single line: '17'
Test #3:
score: -100
Wrong Answer
time: 5ms
memory: 23528kb
input:
10000 10000
output:
982474519
result:
wrong answer 1st lines differ - expected: '689223145', found: '982474519'