Wannafly 挑战赛13 B- Jxc军训

题目描述

在文某路学车中学高一新生军训中，Jxc正站在太阳下站着军姿，对于这样的酷热的阳光，Jxc 表示非常不爽。

Jxc将天空看做一个n*n的矩阵，此时天上有m朵云，这些云会随机分布在m个不同的位置，同时太阳会随机出现在一个位置，Jxc想知道他被太阳晒到的概率是多少，由于他仍在站军姿，所以这个有趣的问题就交给了你。考虑到精度问题，Jxc只需要知道这个概率在对998244353取模意义下的值。

Tips：一个分数p/q在模意义下的值即p*q^-1在模意义下的值，X^p-11 (mod p)

输入描述:

输入只有一行，包含两个整数n、m。n和m的意义见题面.

输出描述:

第一行包含一个整数Ans，为答案

示例1

输入

2 2

输出

499122177

备注:

1 <= n, m <= 2000,m <=n^2

解题思路：如果你求你被太阳晒到的几率就是（n*n-m)/(n*n),然后对998244353取模，这时候我们需要用到的是费马小定理里的一个定理a^(p-1) 恒等于1，要求是 a,p互质并且为素数。 那么a^(p-2)=1/a. 然后我们再用一下快速幂求分母就行了。

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const long long mo=998244353;
long long n,m;
long long find(long long a,long long t){
	long long res=1;
	while(t){
		if(t&1) res=(res*a%mo)%mo;
		a=(a*a)%mo;
		t/=2;
	}
	return res;
} 
int main(){
	int i,j;
	scanf("%lld%lld",&n,&m);
	long long k=(n*n-m)%mo;
	k=k*find(n*n,mo-2)%mo;
	printf("%lld\n",k);
	return 0;
}