51nod 1352 集合计数(扩展欧几里得)

题目链接：传送门

题意：略

分析：

很简单可以得到一个方程 A*x + B*y = N + 1

这式子可以用扩展GCD求出gcd，x和y，然后我们求出大于0的最小x，A*x第一个满足条件的集合firstSet，剩下的N-firstSet个集合可以直接除LCM(A,B)(A和B的最小公倍数)统计出数量。

代码如下：

#include <stdio.h>
#include <string.h>
#include <iostream>
#define LL long long
using namespace std;
LL exgcd(LL a, LL b, LL &x, LL &y) {
    LL r,t;
    if(b==0) {
        x=1;
        y=0;
        return a;
    }
    r=exgcd(b,a%b,x,y);
    t=x;
    x=y;
    y=t-a/b*y;
    return r;
}
int main() {
    int t;
    scanf("%d", &t);
    while(t--) {
        LL ans = 0;
        LL N, A, B;
        LL xx,yy,d,r;
        scanf("%I64d%I64d%I64d",&N, &A, &B);
        d=exgcd(A, B, xx, yy);
        if(((N + 1) % B) % d != 0) ans = 0;
        else {
            LL lcm = A * B / d;
            xx = xx *(((N + 1) % B) / d);
            r = B / d;
            xx=(xx % r + r) % r;
            if(xx == 0) {
                xx = lcm / A;
            }
            if(xx * A > N) {
                ans = 0;
                printf("%I64d\n", ans);
                continue;
            }
            ans += ((N - xx * A) / lcm);
            ans++;
        }
        printf("%I64d\n", ans);
    }
    return 0;
}