牛客小白月赛98（C 两个函数）

kingo_

已于 2024-07-12 21:16:07 修改

阅读量111

点赞数 2

文章标签： c++ 算法

于 2024-07-12 21:09:49 首次发布

本文链接：https://blog.csdn.net/2301_81235183/article/details/140388591

版权

这个题读了好久题意才明白他想表达什么

其实是 g（1）=f（1）=a

g（2）=f（ f (1) ）= a *a

然后g（3）=（ f ( 1 )+f ( 2) ）*a = 3*a *a

g(n) = ( f(1)+ .........+f(n) ) *a = n(n-1)/2 *a;

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int MOD = 998244353,N=1008611;
int a[N],cnt,d,n;
signed main()
{
	ios::sync_with_stdio(0);cin.tie(0);
	int q;cin>>q;
	while(q--)
	{
		 int a,x;
		 cin>>a>>x;
		 int f1=a%MOD;
        int f2=a*a%MOD;
        int res=0;
        if(x==1){cout<<f1<<"\n";continue;}
        if(x==2){cout<<f2<<"\n";continue;}
        int t = x*(x-1)/2%MOD;
        res = t*f2%MOD;
		 cout<<res%MOD<<"\n";
	}
}