Code Forces 582 A. GCD Table(数论)

Description
由n个数,可以构成n*n的gcd矩阵,无序的输入这n*n个数,然后让你找到这n个数,满足这个gcd矩阵
Input
第一行为一整数n表示数的个数,第二行n*n个整数表示这n个数gcd矩阵里的所有元素值
Output
输出满足输入的gcd矩阵的n个数
Sample Input
4
2 1 2 3 4 3 2 6 1 1 2 2 1 2 3 2
Sample Output
4 3 6 2
Solution
最大的数字肯定是满足条件的数,我们就选择最大的数,然后让这个数和前面取到的数,都会产生两个gcd,于是就在数列中把这两个gcd删去就好了,每两个数之间,一定会产生两个gcd的,但是直接用数组记录数字会爆所以用map
Code

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<map>
#include<vector>
using namespace std;
#define maxn 500*500+1
int a[maxn];
map<int,int>m;
vector<int>ans;
int gcd(int a,int b)
{
    if(a<b)return gcd(b,a);
    if(b==0)return a;
    return gcd(b,a%b);
}
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        ans.clear();
        for(int i=1;i<=n*n;i++)
        {
            scanf("%d",&a[i]);
            m[a[i]]++;
        }
        sort(a+1,a+n*n+1);
        for(int i=n*n;i>0;i--)
        {
            if(m[a[i]]==0)continue;
            m[a[i]]--;
            for(int j=0;j<ans.size();j++)
            {
                int temp=gcd(ans[j],a[i]);
                m[temp]-=2;
            }
            ans.push_back(a[i]);
        }
        for(int i=0;i<ans.size();i++)
            printf("%d ",ans[i]);
        printf("\n"); 
    }
    return 0;
}

你可能感兴趣的:(Code Forces 582 A. GCD Table(数论))