提出 #74694191

---

ソースコード 拡げる

#include <bits/stdc++.h>
using namespace std;
int main() {
    int N;
    cin >> N;
    int M;
    cin >> M;
    int64_t ans = 0;
    vector<int64_t> A(N);
    for (int i = 0; i < N; i++) {
      cin >> A.at(i);
    }
    vector<int64_t> B(M);
    for (int i = 0; i < M; i++) {
      cin >> B.at(i);
    }
    vector<int64_t> sumA(N);
    sumA.at(0) = A.at(0);
    for (int i = 0; i < N-1; i++) {
      sumA.at(i+1) = (sumA.at(i) + A.at(i+1));
    }
    vector<int64_t> sumB(M);
    sumB.at(0) = B.at(0);
    for (int i = 0; i < M-1; i++) {
      sumB.at(i+1) = (sumB.at(i) + B.at(i+1));
    }
    for (int i = 0; i < N; i++) {
      ans += ((i+1)*A.at(i)%998244353*sumB.at(M-1)%998244353);
      ans %= 998244353;
    }
    ans -= (sumA.at(N-1)*B.at(0)*(1));
    ans %= 998244353;
    for (int i = 0; i < N-1; i++) {
      ans -= ((sumA.at(N-1)-sumA.at(i))%998244353*B.at(0)*(1));
      ans %= 998244353;
    }
    for (int j = 1; j < M; j++) {
      int i = j-1;
      int cnt = 1;
      while (cnt <= N/(j+1)) {
        ans -= ((sumA.at(N-1)-sumA.at(i))*B.at(j)%998244353*(j+1)%998244353);
        i += j+1;
        ++cnt;
        ans %= 998244353;
      }
    }
    if (ans < 0) {
      ans += 998244353;
    }
    cout << ans << endl;
    return 0;
  }

提出情報

ジャッジ結果