求助拉格朗日插值,WA0分
查看原帖
求助拉格朗日插值,WA0分
294736
bovine__kebi楼主2020/6/22 22:14

RT,暴力插的,好像是取模的锅(?),但是改不出来啊/kk

#include<bits/stdc++.h>
using namespace std;
const int Mod=998244353;
const int maxn=2*1e3+5;
int a[maxn],b[maxn],n,k;
inline int qpower(int x,int p)
{
    int ans=1;
    x%=Mod;
    while(p)
    {
        if(p&1)ans=1ll*ans*x%Mod;
        x=1ll*x*x%Mod;
        p>>=1;
    }
    return ans;
}
inline int qwq(int a,int b)
{
    return a<b?a-b+Mod:a-b;
}
inline int lagrange()
{
    int sum=0;
    for(int i=1;i<=n;i++)
    {
        int x=1,y=1;
        for(int j=1;j<=n;j++)
        {
            if(i==j)continue;
            x=1ll*x*qwq(k,a[i])%Mod;
            y=1ll*y*qwq(a[i],a[j])%Mod;
        }
        sum=(1ll*sum+(1ll*b[i]*x%Mod*qpower(y,Mod-2)%Mod)%Mod)%Mod;
    }
    return sum%Mod;
}
int main()
{
    scanf("%d %d",&n,&k);
    for(int i=1;i<=n;i++)scanf("%d %d",&a[i],&b[i]);
    printf("%d\n",lagrange());
}
2020/6/22 22:14
加载中...