1352 集合计数 扩展欧几里德算法

http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1352

题目的本意就是要求

ax+by=n+1

a * x + b * y = c，在x >= 1和y >= 1的条件下的解的个数。

①、等于0的情况是：

1、方程无解，

2、最小的解就是(1. 1)，大家都是1倍，如果比c还大，那就是0

3、设最小的解是x1,如果c - a * x1 < b，那也不行。也就是b不够一倍了。

其他的：（设x1是最小的正整数解）

通解就是x0 = x1 + b / (abgcd) * k，（k >= 0）

然后带入去a * x + b * y = c，同样c - a * x1也是要大于等于b才行，因为起码都要一倍。

然后解出的k，就是答案。包括上0，所以是k + 1

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <assert.h>
#define IOS ios::sync_with_stdio(false)
using namespace std;
#define inf (0x3f3f3f3f)
typedef long long int LL;

#include <iostream>
#include <sstream>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <string>
#include <bitset>
LL exgcd(LL a, LL mod, LL &x, LL &y) {
//求解a关于mod的逆元     ★：当且仅当a和mod互质才有用
if (mod == 0) {
x = 1;
y = 0;
return a;//保留基本功能，返回最大公约数
}
LL g = exgcd(mod, a % mod, x, y);
LL t = x;    //这里根据mod==0  return回来后，
x = y;  //x,y是最新的值x2,y2,改变一下，这样赋值就是为了x1=y2
y = t - (a / mod) * y;    // y1=x2(变成了t)-[a/mod]y2;
return g;            //保留基本功能，返回最大公约数
}

bool get_min_number (LL a, LL b, LL c, LL &x, LL &y) { //得到a*x+b*y=c的最小整数解
LL abGCD = __gcd(a, b);
if (c % abGCD != 0) return false; //不能整除GCD的话，此方程无解
a /= abGCD;
b /= abGCD;
c /= abGCD;
LL tx, ty;
exgcd(a, b, tx, ty); //先得到a*x+b*y=1的解，注意这个时候gcd(a,b)=1
x = tx * c;
y = ty * c;  //同时乘上c，c是约简了的。得到了一组a*x + b*y = c的解。
LL haveSignB = b;
if (b < 0) b = -b;     //避免mod负数啊，模负数没问题，模了负数后+负数就GG
x = (x % b + b) % b;  //最小解
if (x == 0) x = b; //避免x = 0不是"正"整数  不要用这个，溢出
y = (c - a * x); // haveSignB;
return true; //true代表可以
}

void work() {
LL a, b, c;
cin >> c >> a >> b;
c++;
if (c % __gcd(a, b) != 0 || a + b > c) {
cout << 0 << endl;
return;
}
LL x1, y1;
get_min_number(a, b, c, x1, y1);
LL t = c - b - a * x1;
if (c - x1 * a < b) {
//        cout << t << endl;
//        cout << c << " " << a << " " << b << endl;
//        while(1);
cout << 0 << endl;
return;
}
t /= a / __gcd(a, b) * b;
cout << t + 1 << endl;
}

int main() {
#ifdef local
freopen("data.txt", "r", stdin);
//    freopen("data.txt", "w", stdout);
#endif
IOS;
int t;
cin >> t;
while (t--) work();
return 0;
}

View Code