乘法取模中出现的易错点

一开始我写的代码是

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

#define int long long
int q,a,x;
const int mo = 998244353;
int fun(){
	if(x==1) return a%mo;
	return (a*a%mo)*((1+x-1)*(x-1)%mo/2)%mo;
}

signed main(){
	cin >> q;
	while(q--){
		cin >> a >> x;
		cout << fun() << endl;
	}
	return 0;
}

但这个是错的

我们正确的写法应该是

int fun(){
	if(x==1) return a%mo;
	return (a*a%mo)*((1+x-1)*(x-1)/2%mo)%mo;
}