牛客月赛49 D-梵

牛客月赛49 D-梵

题解：画出抛物线即可发现这是一个向上凸的抛物线，取点a和点b中间部分的值时，f(x)>0，所以要求S(n)-S(m)的最大值就是求S(n)-S(m-1)或者S(n)-S(m)
那么只需要在-(m-a)(m-b)-(m+1-a)(m+1-b)…-(n-a)(n-b)上找规律即可
大概可以分为三部分:-(m²+(m+1)²+…+n²)、(a+b)(m+m+1+m+2+…+n)、-(ab+ab+…+ab)
因为1²+2²+…+n²=n*(n+1)(2n+1)/6
所以-(m²+(m+1)²+…+n²)=-n*(n+1)(2n+1)/6+(m-1)m(2m-1)/6
(a+b)(m+m+1+m+2+…+n)=(n+m)(m+n)*(n-m+1)/2
(ab+ab+…+ab)=(n-m+1)nm

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <string>
#include <cstring>
#include <vector>
using namespace std;
#define IOS ios::sync_with_stdio(0), cin.tie(0),cout.tie(0)
#define endl "\n"
typedef long long ll;
typedef pair<int ,int> PII;
const int INF=0x3f3f3f3f;
const ll mod=998244353;
const int N=1e5+5;
ll ksm(ll x,ll y)
{
    ll res=1;
    while(y)
    {
        if (y&1) res=(res*x)%mod;
        x=x*x%mod;
        y>>=1;
    }
    return res;
}
int main()
{
    IOS;
	int T;
	cin>>T;
	while(T--)
	{
		ll m,n;
		cin>>m>>n;
		ll res=0;
        if (n<m) swap(n,m); 
		res=-n*(n+1)%mod*(2*n+1)%mod*ksm(6,mod-2)%mod+(m-1)*m%mod*(2*m-1)%mod*ksm(6,mod-2)%mod;
        res%=mod;
        res=(res-(n-m+1)*n%mod*m%mod)%mod;
        res%=mod;
        res=(res+(n+m)%mod*(m+n)%mod*(n-m+1)%mod*ksm(2,mod-2)%mod)%mod;
		cout<<(res%mod+mod)%mod<<endl;
	}
	return 0;
}