XTU新生赛B题 ABK 解题报告

题目描述

ABK是一个比A+B还要简单的题目,给出两个整数A,B,求出A和B的第K大公约数。

输入
第一行是一个整数N（N ≤ 10000），表示样例的个数。 以后每行一个样例，为3个整数A，B，K (1≤A,B≤109 , 1≤K≤10)
输出
每行输出一个整数d,表示A,B的第K大公约数 若没有第K大的公约数则输出-1。
样例输入
7
12 24 1
12 24 2
12 24 3
12 24 4
12 24 5
12 24 6
12 24 7

样例输出
12
6
4
3
2
1
-1

解题思路

首先明确一点：a,b的公约数等价于gcd(a,b)的约数。

即我们首先求出gcd(a,b),然后分析它的第K大约数即可。

我们可以简单想出一个On的算法，从gcd到1扫一遍，扫到第K个gcd的约数就输出即可 但是10^9数据会T。。。

考虑到约数的性质，我们可以把算法的复杂度降为O(根号n).只需要考虑1,2,.....,根号n ,我们其实就已经知道了前n个数的约数信息了.其实，gcd的第K大约数即为gcd/gcd的第K小约数.

从1扫到根号n,我们用ans数组来装其中的假设有cnt个  t的约数。

我们可以看出，若K > 2*cnt则第K大约数肯定不存在。

如果0<K<=cnt 那么第K大约数在根号n之后。

如果cnt<K<=2*cnt 那么第K大约数在根号n之前。

此外注意一下如果根号n是gcd的约数，那么不要重复计算它。

#include <cstdio>
#include <cmath>
#include <algorithm>
using namespace std;
int gcd(int a,int b) {return b==0 ? a:gcd(b,a%b);}
int ans[100010];
int main()
{
    int t;
    scanf("%d",&t);
    while(t--){
        int a,b,k;
        scanf("%d%d%d",&a,&b,&k);
        int t = gcd(a,b);//即考虑t的第K大约数
        int sqr = floor(sqrt(t*1.0)+0.5);
        int cnt = 0;//cnt变量是1到sqr中t约数的个数
        for(int i = 1 ; i <= sqr ; i ++) if(t%i == 0) ans[++cnt] = i;//扫一遍
        bool flag = false;
        if(ans[cnt]*ans[cnt] == t) flag = true;//flag是特判ans[cnt]本身是t的约数这种情况
        if((!flag && k > 2*cnt)||(flag && k > 2*cnt-1)) {//如果flag的话 ans[cnt]只能算一个
            printf("-1\n");
        } else if(k <= cnt) {
            printf("%d\n",t/ans[k]);
        } else {
            if(!flag) printf("%d\n",ans[2*cnt-k+1]);
            else printf("%d\n",ans[2*cnt-k]);
        }
    }
    return 0;
}

详见代码