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;
      |                       ^