【牛客网】Wannafly挑战赛22，A题计数器：裴蜀定理结合扩展欧几里得思路

题目链接：

https://www.nowcoder.com/acm/contest/160/A

 

题目：

有一个计数器，计数器的初始值为0，每次操作你可以把计数器的值加上a1,a2,...,an中的任意一个整数，操作次数不限（可以为0次），问计数器的值对m取模后有几种可能。

 

输入：

第一行两个整数n,m
接下来一行n个整数表示a1,a2,...,an
1≤n≤100
1≤m,a1,a2,...,an≤1000000000

 

输出：

输出一个整数表示答案

 

分析：

代码很简单，主要是思路，先要了解裴蜀定理以及扩展欧几里得算法

裴蜀定理： 设a1,a2,a3......an为n个整数，d是它们的最大公约数，那么存在整数x1......xn使得x1*a1+x2*a2+...xn*an=m 如果要有解，要求m一定是d的倍数（m=0也行）   扩展欧几里得算法： 对于ax+by=gcd(a,b)的式子 x = x0 + b/gcd(a, b) * t x = x0 - a/gcd(a, b) * t (其中t为任意整数)

所以对于这个题目，我们

假设k1为a1取的个数；根据题意题目变成a1*k1+a2*k2+a3*k3+a4*k4+....an*kn=P=k*gcd(a1,a2,a3,...,an)，其中k为任意整数

这就符合了裴蜀定理，设 g=gcd(a1,a2,,,,,,an)

显然 (k*g)%m 的不同个数就是答案,观察之后发现这个式子是有循环节的

因为是有循环节的，所以我们要找出一个循环节里的个数也就是我们最后要的答案，那么思路就是找到一个一样的值，中间相差的就是循环节

显然，因为题目说可以为0次，那就是都不取，0%m一定是0，所以0就是其中一个出现的，而且还会重复出现

那么：(k*g)%m=0

也就是kg-mk1=0，我们可以认为a=g，b=m，x=k，y=-k1，就变成了ax+by=0，因为0也是gcd(a,b)的倍数，所以也是有解的

那么这个x的解，x=x0 + b/gcd(a, b) * t = x0 + m/gcd(g, m) * t

我们可以看得出，x也就是k，当结果都是0对应k其实是由m/gcd(g, m)来控制的，这个就是循环节的个数（两个k之间差的就是循环节个数）

因此最后的答案其中就是求m和a1,a2,a3....,an的最大公约数d，然后m/d就是答案

 

AC代码：

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

int gcd(int a,int b)
{
	return b?gcd(b,a%b):a;
}

int main()
{
	int n,m;
	cin >> n >> m;
	int input;
	int d = m;
	for(int i = 0;i < n;i++)
	{
		cin >> input;
		d = gcd(d,input);
	} 
	printf("%d\n",m/d);
	
	return 0;
}