Submission #74664187

---

Source Code Expand

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10,mod=998244353;
typedef long long ll;
typedef pair<ll,ll> PII;
typedef array<ll,2> a2;
typedef array<ll,3> a3;
ll n,m,k;
ll a[N],b[N];
ll sum[N];
void __(){
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=1;i<=m;i++) cin>>b[i];
    for(int i=1;i<=n;i++) sum[i]=sum[i-1]+a[i];
    ll S=0;
    for(int i=1;i<=m;i++) S=(S+b[i])%mod;
    ll ans1=0,ans2=0;
    for(int i=1;i<=n;i++){
        ll val=a[i]*i%mod*S%mod;
        ans1=(ans1+val)%mod;
    }
    // for(int i=1;i<=n;i++){
    //     for(int j=1;j<=m;j++){
    //         ans2=ans2+a[i]*b[j]%mod*(i/j*j)%mod;
    //         ans2=(ans2+mod)%mod;
    //     }
    // }
    // cout<<ans2<<endl;
    for(int j=1;j<=m;j++){
        for(int i=0;i*j<=n;i++){
            ll l=i*j,r=(i+1)*j-1;
            r=min(r,n);
            ll val=sum[r]-sum[max(0ll,l-1)];
            val%=mod;
            val=val*i%mod;
            val=val*b[j]%mod;
            ans2=(ans2+val*j)%mod;
        }
    }
    cout<<(ans1-ans2+mod)%mod<<endl;
}


int main(){
    ios::sync_with_stdio(0);
    cin.tie(0);
    int _=1;
    // cin>>_;
    while(_--){
        __();
    }
}

Submission Info

Judge Result