hdu1753 X问题 （扩展CRT）

hdu1753 X问题

原题地址：http://acm.hdu.edu.cn/showproblem.php?pid=1573

题意：
T组数据，
求在小于等于N的正整数中有多少个X满足：
X mod a[0] = b[0],
X mod a[1] = b[1],
X mod a[2] = b[2],
…
X mod a[M] = b[M]…

数据范围
0 < a[i] <= 10，0 < N <= 1000,000,000 , 0 < M <= 10

题解：
扩展中国剩余定理。
如何合并两个同余方程?
$x= m_1k_1+a_1$
$x=m_2k_1+a_2$

$m_1k_1+a_1=m_2k_2+a_2$
$(m_1k_1-m_2k_2)=a_2-a_1$
$m_1k_1\equiv(a_2-a_1) \ \ (mod\ m_2)$

$gcd(m_1,m_2)\nmid(a_2-a_1)$时无解

令 $gcd(m_1,m_2)=d$， $a_2-a_1=c$

则$\frac{m_1}{d} k_1 \equiv \frac{c}{d} \ \ mod(\frac{m_2}{d})$
$k_1 \equiv \frac{c}{d}*(\frac{m_1}{d})^{-1} \ \ mod(\frac{m_2}{d})$
令 $k=\frac{c}{d}*(\frac{m_1}{d})^{-1}$
则$k_1 =\frac{m_2}{d}*y+k$

于是 $x= m_1*(\frac{m_2}{d}*y+k )+a_1=\frac{m_1m_2}{d}*y+m_1*k+a_1$

合并为 $x\equiv m_1*k+a_1\ \ \ mod(\frac{m_1m_2}{d})$ ，其中 $k=(\frac{m_1}{d})^{-1}$

代码：

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
#define LL long long
using namespace std;
const int N=20;
int a[N],b[N],n,m,T;
bool flag;
int getgcd(int A,int B) {return B==0?A:getgcd(B,A%B);}
void exgcd(int A,int B,int &x,int &y)
{ 
    if(B==0) {x=1; y=0; return;}
    int x0,y0;
    exgcd(B,A%B,x0,y0);
    x=y0;
    y=x0-(A/B)*y0;
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(int i=1;i<=m;i++) scanf("%d",&a[i]);
        for(int i=1;i<=m;i++) scanf("%d",&b[i]);
        int a1=b[1]; int m1=a[1]; int a2,m2; flag=0;
        for(int  i=2;i<=m;i++)
        {
            a2=b[i]; m2=a[i]; if(a2<a1) {swap(a1,a2); swap(m1,m2);}
            int c=a2-a1; int d=getgcd(m1,m2);
            if(c%d) {flag=1;break;}
            int x,y;  exgcd(m1/d,m2/d,x,y);
            x=(x+(m2/d))%(m2/d); x=(x*(c/d)+(m2/d))%(m2/d);
            a1=m1*x+a1; m1=(m1*m2)/d;
        }
        if(flag||a1>n) printf("%d\n",0);
        else if(a1==0) printf("%d\n",(n-a1)/m1);
        else printf("%d\n",(n-a1)/m1+1);
    }
    return 0;
}