codeforces #323 c GCD Table(map)

題目連接:http://codeforces.com/contest/583/problem/C

題意:由n個數,可以構成n*n的gcd矩陣,無序的輸入這n*n個數,然後讓你找到這n個數,滿足這個gcd矩陣

思路:最大的數字肯定是滿足條件的數,我們就選擇最大的數,然後讓這個數和前面取到的數,都會產生兩個gcd,於是就在數列中把這兩個gcd刪去就好了。每兩個數之間,一定會產生兩個gcd的

本題最大的問題就是,用h[a[i]]記錄數字時,數組即使開10的9次都不夠,get到的新技能就是用map,別的用途不是很會,但是他可以開的數組很大很大

代碼:

#include <iostream>
#include <algorithm>
#include <cmath>
#include<map>
#include<vector>
#include <cstring>
#include <cstdio>
#include <iomanip>
#include <queue>
#define ll long long
#define maxn 490000
using namespace std;
map <int,int> h;
ll a[maxn];
vector<int>ans;
int gcd(int a,int b)
{
   if(b==0)
    return a;
   return gcd(b,a%b);
}
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        ans.clear();
        //memset(h,0,sizeof(h));
        for(int i=0;i<n*n;i++)
        {
            scanf("%I64d",&a[i]);
            h[a[i]]++;
        }
        sort(a,a+n*n);
        for(int i=n*n-1;i>=0;i--)
        {
            if(h[a[i]]==0)
                continue;
            h[a[i]]--;
            for(int j=0;j<ans.size();j++)
            {
                int k=gcd(a[i],ans[j]);
                h[k]-=2;
            }
            ans.push_back(a[i]);
        }
        for(int i=0;i<ans.size();i++)
            printf("%d ",ans[i]);
        printf("\n");
    }
    return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章