51Nod - 1352 拓展gcd

题意：

给出N个固定集合{1，N},{2,N-1},{3,N-2},...,{N-1,2},{N,1}.求出有多少个集合满足：第一个元素是A的倍数且第二个元素是B的倍数。

提示：

对于第二组测试数据，集合分别是：{1,10},{2,9},{3,8},{4,7},{5,6},{6,5},{7,4},{8,3},{9,2},{10,1}.满足条件的是第2个和第8个。

Input

第1行：1个整数T（1<=T<=50000)，表示有多少组测试数据。
第2 - T+1行：每行三个整数N，A,B（1<=N，A，B<=2147483647)

Output

对于每组测试数据输出一个数表示满足条件的集合的数量，占一行。

Input示例

2
5 2 4
10 2 3

Output示例

1
2

思路：

根据题意可以列出一个不定方程a*x+b*y=n+1，利用拓展gcd求一组可行解。

要求出大于0的最小x，直接用所求出的一组可行解(x%(b/g) + (b/g))%(b/g)即可(其中g是a和b的最大公约数)。

这样就求得了最小的a*x，那么剩下的就是n-a*x个数，x的通解形式是x+k*(b/g)，那么a*x的通解就是a*x+k*(a*b/g)==a*x+k*lcm，所以直接拿n-a*x除以lcm(a,b)再算上一开始的那个最小的a*x就是最终答案。

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

ll extgcd(ll a, ll b, ll &x, ll &y) {
    ll d = a;
    if (b != 0) {
        d = extgcd(b, a % b, y, x);
        y -= (a / b) * x;
    }
    else {
        x = 1; y = 0;
    }
    return d;
}

int main() {
    int T;
    scanf("%d", &T);
    while (T--) {
        ll n, a, b, x, y;
        scanf("%I64d%I64d%I64d", &n, &a, &b);
        ++n;
        ll g = extgcd(a, b, x, y);
        if (n % g) {
            puts("0");
            continue;
        }
        x *= n / g;
        ll bb = b / g;
        x = (x % bb + bb) % bb;
        if (x == 0) x += bb;
        if (a * x >= n) {
            puts("0");
            continue;
        }
        else {
            ll lcm = a / g * b;
            printf("%I64d\n", (n - 1 - x * a) / lcm + 1);
        }
    }
    return 0;
}