求大佬指点最后样例没过

#include<bits/stdc++.h>
using namespace std;
long long sum;
const int MOD=998244353;
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        long long a,x;
        cin>>a>>x;
        if(x==1)
        {
            cout<<a*x%998244353<<endl;
        }
         else{
             for(int i=1;i<=x-1;i++)
        {
          sum=sum%MOD+(a*a)%MOD*(i%MOD);
        }
       cout<<sum%MOD<<endl;
         }
    }
   
}