模板

已知 (n+1) 个点，拟合一个多项式 (f) ，求 (f(x)) 的值。

拉格朗日插值多项式为： (f(x)=sumlimits_{i=0}^{n}y_ifrac{prod_{j=0,j eq i}^{n}(x-x_j)}{prod_{j=0,j eq i}^{n}(x_i-x_j)})

从数组 (x) 和 (y) 的 ([0,n]) 共 (n+1) 个点，插值出一个 (n) 次多项式 (f) ，然后求出 (f(x_i)) 返回。

int lagrange(int n, int *x, int *y, int xi) {
    int ans = 0;
    for (int i = 0; i <= n; i++) {
        int s1 = 1, s2 = 1;
        for (int j = 0; j <= n; j++)
            if (i != j) {
                s1 = 1ll * s1 * (xi - x[j]) % mod;
                s2 = 1ll * s2 * (x[i] - x[j]) % mod;
            }
        ans = (1ll * ans + 1ll * y[i] * s1 % mod * quick_pow(s2, mod - 2) % mod) % mod;
    }
    return (ans + mod) % mod;
}

从数组 (x) 和 (y) 的 ([0,n]) 共 (n+1) 个点，插值出一个 (n) 次多项式 (f) ，然后求出 (f(x_i)) 返回，其中数组 (x) 的点需要是连续自然数。

int lagrange(int n, int *x, int *y, int xi) {
    int ans = 0;
    s1[0] = (xi - x[0]) % mod, s2[n + 1] = 1;
    for (int i = 1; i <= n; i++)
        s1[i] = 1ll * s1[i - 1] * (xi - x[i]) % mod;
    for (int i = n; i >= 0; i--)
        s2[i] = 1ll * s2[i + 1] * (xi - x[i]) % mod;
    ifac[0] = ifac[1] = 1;
    for (int i = 2; i <= n; i++)
        ifac[i] = -1ll * mod / i * ifac[mod % i] % mod;
    for (int i = 2; i <= n; i++)
        ifac[i] = 1ll * ifac[i] * ifac[i - 1] % mod;
    for (int i = 0; i <= n; i++)
        (ans += 1ll * y[i] * (i == 0 ? 1 : s1[i - 1]) % mod * s2[i + 1] % mod
                * ifac[i] % mod * (((n - i) & 1) ? -1 : 1) * ifac[n - i] % mod) %= mod;
    return (ans + mod) % mod;
}