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

题目传送门

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

代码：

#include<bits/stdc++.h>
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);
}