Submission #74690668

Source Code Expand

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const int mOD = 998244353;

int main() {
    int n, m;
    cin >> n >> m;
    vector<ll> 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<ll> S1(n + 1), S2(n + 1);
    for (int i = 1; i <= n; ++i) {
        S1[i] = (S1[i - 1] + A[i] * i % mOD) % mOD;
        S2[i] = (S2[i - 1] + A[i]) % mOD;
    }
    ll total_S1 = S1[n];

    ll ans = 0;
    for (int j = 1; j <= m; ++j) {
        ll bj = B[j] % mOD;
        ll sum = 0;
        for (int l = 1; l <= n; l += j) {
            int r = min(l + j - 1, n);
            ll cnt = (l - 1) / j;
            ll sumA = (S2[r] - S2[l - 1] + mOD) % mOD;
            sum = (sum + cnt * sumA % mOD) % mOD;
        }

        ll term = (total_S1 - j % mOD * sum % mOD + mOD) % mOD;
        ans = (ans + bj * term % mOD) % mOD;
    }

    cout << ans << endl;
    return 0;
}

Submission Info

Submission Time
Task E - You WILL Like Sigma Problem
User wuqize
Language C++23 (GCC 15.2.0)
Score 0
Code Size 990 Byte
Status WA
Exec Time 211 ms
Memory 18884 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 0 / 450
Status
WA × 2
WA × 22
Set Name Test Cases
Sample 00-sample-01.txt, 00-sample-02.txt
All 00-sample-01.txt, 00-sample-02.txt, 01-01.txt, 01-02.txt, 01-03.txt, 01-04.txt, 01-05.txt, 01-06.txt, 01-07.txt, 01-08.txt, 01-09.txt, 01-10.txt, 01-11.txt, 01-12.txt, 01-13.txt, 01-14.txt, 01-15.txt, 01-16.txt, 01-17.txt, 01-18.txt, 01-19.txt, 01-20.txt
Case Name Status Exec Time Memory
00-sample-01.txt WA 1 ms 3520 KiB
00-sample-02.txt WA 1 ms 3456 KiB
01-01.txt WA 1 ms 3404 KiB
01-02.txt WA 1 ms 3568 KiB
01-03.txt WA 1 ms 3556 KiB
01-04.txt WA 3 ms 3684 KiB
01-05.txt WA 3 ms 3712 KiB
01-06.txt WA 3 ms 3744 KiB
01-07.txt WA 3 ms 3780 KiB
01-08.txt WA 3 ms 3808 KiB
01-09.txt WA 211 ms 18884 KiB
01-10.txt WA 210 ms 18844 KiB
01-11.txt WA 89 ms 7236 KiB
01-12.txt WA 88 ms 7152 KiB
01-13.txt WA 89 ms 7172 KiB
01-14.txt WA 106 ms 9132 KiB
01-15.txt WA 210 ms 18808 KiB
01-16.txt WA 210 ms 18864 KiB
01-17.txt WA 210 ms 18864 KiB
01-18.txt WA 210 ms 18840 KiB
01-19.txt WA 90 ms 15068 KiB
01-20.txt WA 134 ms 15612 KiB