[牛客月赛][初级][49]-D题梵快速幂求逆元题解

NHS6671

已于 2022-05-10 18:53:48 修改

阅读量843

点赞数

分类专栏：牛客月赛文章标签：算法 c++ 职场和发展

于 2022-05-10 16:59:39 首次发布

本文链接：https://blog.csdn.net/NHS6671/article/details/124690774

版权

牛客月赛专栏收录该内容

2 篇文章 0 订阅

订阅专栏

题目

样例输入

3
2 4
2 4
2 4

样例输出

1
1
1

题意理解

总结一下题意其实就是函数   $f(x)= -(x-a)(x-b)$

$S(i)=\sum_{x=1}^{i}f(x)=f(1)+f(2)+f(3)+...+f(i)$

而 $f(x)$ 的图像大致如图所示

那么 $S(m)-S(n)$ 的最大值其实就是两个零点的相减，也就是 $S(b)-S(a)$

现在我们得出结论以后再回来看 $S(i)$ 怎么求出来

我们先来看 $f(x)$ 单个是什么我们把 $f(x)$ 打开就是如下这个式子

$f(x)=-(x^{2}-(a+b)\ast x +a\ast b)$

再化简一下就是

$f(x)=-x^{2} +(a+b)\ast x -a\ast b$

那么知道了以后我们来看 $S(n)$ 怎么求

$S(n)= \sum_{i=1}^{n} f(x)=-\sum_{i=1}^{n} x^{2} +(a+b) \sum_{i=1}^{n} x-a\ast b \sum_{i=1}^{n}1$

我们将上式划分为3部分逐一分析

第一部分： $\sum_{i=1}^{n} x^{2}$

这一部分我们可以通过等差数列求和公式总结出来就是 $\frac{n*(n+1)*(2*n+1)}{6}$

第二部分： $\sum_{i=1}^{n} x$

这一部分我们可以通过等差数列求和公式总结出来就是   $\frac{n*(n+1)}{2}$

第三部分： $\sum_{i=1}^{n} 1$

这一部分其实就是n个1加起来那么其实就是n

那么再将这些总结带回原式 $S(n)$ 即可求出

还有就是快速幂求逆元，在mod一个数的情况下除以一个数等于乘以一个数的逆元

在c++中除并不好处理所以我们用乘以一个数的逆元，在本题中就是乘以2和6的逆元

为了避免我们在MOD过程中

MOD出来了负数

那么我们先MOD再加MOD再MOD就一定是一个正数

这是MOD的一个小技巧

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const LL MOD=998244353;
//快速幂
LL qmi(LL a,LL b,LL p){
    LL ans=1;
    while(b){
        if(b&1)ans=ans*a%p;
        a=a*a%p;
        b>>=1;
    }
    return ans;
}
LL a,b;
//逆元
LL inv(LL x){
   return (qmi(x,MOD-2,MOD)%MOD+MOD)%MOD;
}
LL sum(LL x){
    LL ans1=x*(x+1)%MOD*(2*x+1)%MOD*inv((LL)6)%MOD;//第一部分
    LL ans2=(a+b)%MOD*x%MOD*(x+1)%MOD*inv((LL)2)%MOD;//第二部分
    LL ans3=a*b%MOD*x%MOD;//第三部分
    LL ans=((-ans1+ans2-ans3)%MOD+MOD)%MOD;
    return ans;
}
void solve(){
     cin>>a>>b;
     LL ans=(( sum(b)-sum(a) )%MOD+MOD)%MOD;
     cout<<ans<<endl;
}
int main(){
    int T;
    cin>>T;
    while(T--){
        solve();
    }
    return 0;
}