洛谷 P4781 【模板】拉格朗日插值

题目传送门

参考:拉格朗日插值 学习笔记

代码:

#include
using namespace std;

typedef long long ll;
const int maxn=2000+100;
const int mod=998244353;

int x[maxn],y[maxn]; 

ll mypow(ll a,ll b){
	
	ll sum=1;
	while(b){
		
		if(b&1) sum=sum*a%mod;
		a=a*a%mod;
		b>>=1;
	} 
	return sum;
}

int main(){
	
	int n,k;
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++) scanf("%d%d",&x[i],&y[i]);
	ll sum=0;
	for(int i=1;i<=n;i++){
		
		ll fz=y[i],fm=1;
		for(int j=1;j<=n;j++){
			
			if(j==i) continue;
			fz=(fz*(k-x[j])%mod+mod)%mod;
			fm=(fm*(x[i]-x[j])%mod+mod)%mod;
		}
		fm=mypow(fm,mod-2);
		sum=(sum+fz*fm%mod)%mod;
	}
	printf("%lld\n",sum);
}

 

你可能感兴趣的:(ACM,洛谷)