GDUT专题练习4 Happy 2006 数论

题目链接:2022_GDUT_新生专题训练_数论 - Virtual Judge (vjudge.net)

题目大意:

对于给定的m(1<=m<=1e6),k(1<=k<=1e8),求出第k个与m互质的数

思路:

直接暴力复杂度为O(K),显然不可取

注意到m最大为1e6，那么出题人可能出的复杂度是O(nlogn)或者O(n)

我们看这样一条公式gcd(a,m)=gcd(a+m*t,m)

也就是说与m互质的数也是有类似周期性的

那么我们只要枚举小于m的与m互质的数,存放在数组ans中，总数为cnt

如果k<=cnt,那么可以直接出答案ans[k]

如果k>cnt,那么答案为(k-1)/cnt*m+ans[k%cnt]

代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cctype>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<ctime>
#include<vector>
#include<set>
#include<map>
#include<stack>
using namespace std;
typedef long long uu;
uu ans[(uu)1e6+5000];

// gcd(a,m)=gcd(a+m*t,m)


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

int main()
{
	//freopen("input.txt","r",stdin);
	int m,k,i,j,cnt;
	while(cin>>m>>k)
	{
		cnt=0;
		for(i=1; i<=m; i++)
		{
			if(gcd(i,m)==1)
				ans[++cnt]=i;
		}
        ans[0]=ans[cnt];//存在k%cnt=0的情况
		if(k<=cnt)
			cout<<ans[k]<<endl;
		else
			cout<<(k-1)/cnt*m+ans[k%cnt]<<endl;
	}
}