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;
}