拉格朗日插值学习笔记

P4781 【模板】拉格朗日插值

方法 3：拉格朗日插值法

根据多项式除法那么我们会有：

$$f(x)\equiv f(a)(\mathrm{m}\mathrm{o}\mathrm{d}(x-a))$$

这是显然的，因为 $f(x)-f(a)=(a_0-a_0)+a_1(x^1-a^1)+a_1(x^2-a^2)+\cdots +a_n(x^n-a^n)$，这个式子显然有 $(x-a)$ 这个因式，所以得证。

这样我们就可以列一个关于 $f(x)$ 的多项式线性同余方程组：

$$\{\begin{array}{l}f(x)\equiv y_1(\mathrm{m}\mathrm{o}\mathrm{d}(x-x_1))\\ f(x)\equiv y_n(\mathrm{m}\mathrm{o}\mathrm{d}(x-x_2))\\ \cdots \\ f(x)\equiv y_n(\mathrm{m}\mathrm{o}\mathrm{d}(x-x_n))\end{array}$$

我们根据中国剩余定理，有：

$$M=\prod _{i=1}^n(x-x_i),m_i=\frac{M}{x-x_i}=\prod _{j\ne i}(x-x_j)$$

则 $m_i$ 模 $(x-x_i)$ 意义下的逆元就是：

$$m_i^{-1}=\prod _{j\ne i}\frac{1}{x_i-x_j}$$

所以就有：

$$f(x)\equiv \sum _{i=1}^n{y}_i{m}_i{m}_i^{-1}\equiv \sum _{i=1}^n{y}_i\prod _{j\ne i}\frac{x-x_j}{x_i-x_j}(\mathrm{m}\mathrm{o}\mathrm{d}M)$$

所以在模意义下 $f(x)$ 就是唯一的，即：

$$f(x)=\sum _{i=1}^n{y}_i\prod _{j\ne i}\frac{x-x_j}{x_i-x_j}$$

这就是拉格朗日插值的表达式。

如果要将每一项的系数都算出来，时间复杂度仍为 $O(n^2)$，但是本题中只用求出 $f(k)$ 的值，所以在计算上式的过程中直接将 $k$ 代入即可。

$$f(k)=\sum _{i=1}^n{y}_i\prod _{j\ne i}\frac{k-x_j}{x_i-x_j}$$

本题中，还需要求解逆元。如果先分别计算出分子和分母，再将分子乘进分母的逆元，累加进最后的答案，时间复杂度的瓶颈就不会在求逆元上，时间复杂度为 $O(n^2)$。

代码实现

#include<bits/stdc++.h>
#include <unordered_map>
using namespace std;
template<class...Args>
void debug(Args... args) {//Parameter pack
	auto tmp = { (cout << args << ' ', 0)... };
	cout << "\n";
}
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll, ll>pll;
typedef pair<int, int>pii;
const ll N = 1e5 + 5;
const ll INF = 0x7fffffff;
const ll MOD = 1e9 + 7;

ll qpow(ll x, ll n, ll mod) {
	ll ans = 1;
	while (n) {
		if (n & 1)ans = (ans * x) % mod;
		x = (x * x) % mod;
		n >>= 1;
	}
	return ans % mod;
}
ll get_inv(ll num,ll mod) {
	return qpow(num, mod - 2, mod);
}
ll lagrange(vector<ll> x,vector<ll> y,int k ,int mod) {
	int n = x.size();
	ll ans = 0;
	ll s1, s2;
	for (int i = 0; i < n; i++) {
		s1 = y[i] % mod;
		s2 = 1;
		for (int j = 0; j < n; j++) {
			if (i != j) {
				s1 = s1 * (k - x[j]) % mod;
				s2 = s2 * ((x[i] - x[j] % mod) % mod) % mod;
			}
		}
		ans += s1 * get_inv(s2, mod) % mod;
		ans = (ans + mod) % mod;
	}
	return ans;
}
int main() {
	ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);

	int n, k;
	cin >> n >> k;
	vector<ll>x(n),y(n);
	for (int i = 0; i < n; i++) cin >> x[i] >> y[i];
	cout << lagrange(x,y,k, 998244353);

	return 0;
}