中国剩余定理及其拓展

中国剩余定理

求下列同余方程组的解（保证$n_i,n_j$两两互质）：

$x\equiv a_1(\mathrm{m}\mathrm{o}\mathrm{d}{n}_1)$
$x\equiv a_2(\mathrm{m}\mathrm{o}\mathrm{d}{n}_2)$
$x\equiv a_3(\mathrm{m}\mathrm{o}\mathrm{d}{n}_3)$
…
$x\equiv a_k(\mathrm{m}\mathrm{o}\mathrm{d}{n}_k)$

唯一解：$ans\equiv \sum a_i\ast c_i(\mathrm{m}\mathrm{o}\mathrm{d}M)$，其中$M=\prod n_i,m_i=M/n_i,m_i^{-1}$为$m_i$在$n_i$意义下的逆元，$c_i=m_i\ast m_i^{-1}$（不对n_i取模）。

证明：

1. $i\ne j$时，因为$m_i=M/n_i$，所以$m_i\mathrm{m}\mathrm{o}\mathrm{d}{n}_j=0$，即$c_i\equiv m_i\equiv 0(\mathrm{m}\mathrm{o}\mathrm{d}{n}_j)$
2. $ans\equiv {\sum }_{j=1}^k{a}_j\ast c_j(\mathrm{m}\mathrm{o}\mathrm{d}{n}_i)$
3. $ans\equiv a_i\ast c_i(\mathrm{m}\mathrm{o}\mathrm{d}{n}_i)$
4. 因为$c_i=m_i\ast m_i^{-1}$,所以$c_i\equiv 1(\mathrm{m}\mathrm{o}\mathrm{d}{n}_i)$
5. $ans\equiv a_i(\mathrm{m}\mathrm{o}\mathrm{d}{n}_i)$

代码：

int a[N],b[N],m[N],c[N];
void exgcd(int a,int b,int &x,int &y)
{
	if(!b)
	{
		x = 1;
		y = 0;
		return ;
	}
	exgcd(b,a%b,y,x);
	y-=(a/b)*x;
}

int crt(int n)
{
	int mul = 1,res = 0;
	for(int i=1;i<=n;i++)
		mul *= a[i];
	for(int i=1;i<=n;i++)
		m[i] = mul/a[i];
	for(int i=1;i<=n;i++)
	{
		int x,y;
		exgcd(m[i],a[i],x,y);
		c[i] = m[i]*x;
		res = (res + b[i]*c[i])%mul;
	}
	return (res+mul)%mul;
}
signed main()
{
	int n;
	cin>>n;
	for(int i=1;i<=n;i++) cin>>a[i]>>b[i];
	cout<<crt(n)<<endl;
	return 0;
}

拓展中国剩余定理：

求下列同余方程组的解（不保证$n_i,n_j$两两互质）：

$x\equiv a_1(\mathrm{m}\mathrm{o}\mathrm{d}{n}_1)$
$x\equiv a_2(\mathrm{m}\mathrm{o}\mathrm{d}{n}_2)$
$x\equiv a_3(\mathrm{m}\mathrm{o}\mathrm{d}{n}_3)$
…
$x\equiv a_k(\mathrm{m}\mathrm{o}\mathrm{d}{n}_k)$

先看前两个方程:

1. $x=k_1\ast n_1+a_1=k_2\ast n_2+a_2$
2. $a_2-a_1=k_1\ast n_1-k_2\ast n_2$
3. 当$(a_2-a_1)\mathrm{m}\mathrm{o}\mathrm{d}gcd(n_1,n_2)\ne 0$时无解
4. 有解时，通过exgcd算法求出$k_1,k_2$的一组解
5. 回代$k_1$,即有$x\equiv k_1\ast n_1+a_1(\mathrm{m}\mathrm{o}\mathrm{d}lcm(n_1,n_2))$
6. 多次求解，合并

代码：

int a[N],b[N];
void exgcd(int a,int b,int &x,int &y)
{
	if(!b)
	{
		x = 1;
		y = 0;
		return ;
	}
	exgcd(b,a%b,y,x);
	y-=(a/b)*x;
}
int qmul(ll a,ll b,ll p)
{
	ll res = 0;
	while(b)
	{
		if(b&1)
			res = (res + a) % p;
		a = (a+a)%p;
		b>>=1;
	}
	return res;
}
signed main()
{
	int n;
	cin>>n;
	for(int i=1;i<=n;i++) cin>>a[i]>>b[i];
	int ans = b[1],M = a[1];
	for(int i=2;i<=n;i++)
	{
		int x,y;
		int c = ((b[i]-ans)%a[i]+a[i])%a[i];
		exgcd(M,a[i],x,y);
		int d = __gcd(M,a[i]);
		x = qmul(x,c/d,a[i]);
		ans+=M*x;
		M*=a[i]/d;
		ans = (ans+M)%M;
	}
	cout<<ans<<endl;
	return 0;
}