Submission #74685239
Source Code Expand
#include <bits/stdc++.h>
using namespace std;
const int MOD = 998244353;
typedef long long ll;
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
int N, M;
cin >> N >> M;
vector<int> A(N + 1), B(M + 1);
for (int i = 1; i <= N;i++) cin >> A[i];
for (int j = 1; j <= M;j++) cin >> B[j];
vector<int> preA(N + 1, 0);
ll sumA_i_i = 0; // Σ A[i] * i
for(int i=1;i<=N;i++)preA[i]=(preA[i - 1] + A[i]) % MOD,sumA_i_i = (sumA_i_i + (ll)A[i] * i) % MOD;
ll sumB = 0;
for (int j = 1; j <= M;j++)sumB = (sumB + B[j]) % MOD;
ll part1 = sumA_i_i * sumB % MOD,ll T = 0,Jmax = min(M, N);
for (int j = 1; j <= Jmax;j++) {
ll Sj = 0;
int maxk = N / j;
for (int k = 1; k <= maxk;k++) {
int L = k * j;
int R = min(N, (k + 1) * j - 1);
int seg = (preA[R] - preA[L - 1] + MOD) % MOD;
Sj = (Sj + (ll)k * seg) % MOD;
}
T = (T + (ll)B[j] * j % MOD * Sj) % MOD;
}
ll ans = (part1 - T + MOD) % MOD;
cout << ans << '\n';
return 0;
}
Submission Info
| Submission Time | |
|---|---|
| Task | E - You WILL Like Sigma Problem |
| User | Atcoder__ |
| Language | C++23 (GCC 15.2.0) |
| Score | 0 |
| Code Size | 1145 Byte |
| Status | CE |
Compile Error
./Main.cpp: In function 'int main()':
./Main.cpp:17:55: error: 'MOD\U0000ff1b' was not declared in this scope
17 | for (int j = 1; j <= M;j++)sumB = (sumB + B[j]) % MOD;
| ^~~~~
./Main.cpp:19:26: error: 'Jmax' was not declared in this scope; did you mean 'fmax'?
19 | for (int j = 1; j <= Jmax;j++) {
| ^~~~
| fmax
./Main.cpp:28:9: error: 'T' was not declared in this scope
28 | T = (T + (ll)B[j] * j % MOD * Sj) % MOD;
| ^
./Main.cpp:30:15: error: 'part1' was not declared in this scope
30 | ll ans = (part1 - T + MOD) % MOD;
| ^~~~~
./Main.cpp:30:23: error: 'T' was not declared in this scope
30 | ll ans = (part1 - T + MOD) % MOD;
| ^