CodeForces 582A GCD Table (分析狀态+貪心)

題目連結:http://codeforces.com/problemset/problem/582/A

題目大意

給定一個序列的GCD table的亂序,

GCDtable的定義十分簡單形象,

要求根據這個table 序列還原出原來的序列.

題目分析 

我們可以發現,table中最大的一個數肯定是序

列中的一個數,這樣狀态就可以根據目前情況縮減了!!

假設我們已經得知前k個數,那麼對于目前table 序列,

最大數肯定是剩餘序列中的一個數,但還要消去

這個數和之前的數對目前序列的影響,

于是模拟這樣的過程,想到用一個vector存儲已經選出來的數,

再用map計數下即可.

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

#define debug puts("YES");
#define rep(x,y,z) for(int (x)=(y);(x)<(z);(x)++)
#define ll long long

#define lrt int l,int r,int rt
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define root l,r,rt
#define mst(a,b) memset((a),(b),sizeof(a))
#define pii pair<int,int>
#define fi first
#define se second
#define mk(x,y) make_pair(x,y)
const int mod=1e9+7;
const int maxn=5e2+10;
const int ub=1e6;
ll powmod(ll x,ll y){ll t; for(t=1;y;y>>=1,x=x*x%mod) if(y&1) t=t*x%mod; return t;}
int gcd(int x,int y){
    if(y==0) return x;
    return gcd(y,x%y);
}
/*
題目大意:
給定一個序列的GCD table的亂序,
GCDtable的定義十分簡單形象,
要求根據這個table 序列還原出原來的序列.

題目分析:
我們可以發現,table中最大的一個數肯定是序
列中的一個數,這樣狀态就可以根據目前情況縮減了!!
假設我們已經得知前k個數,那麼對于目前table 序列,
最大數肯定是剩餘序列中的一個數,但還要消去
這個數和之前的數對目前序列的影響,
于是模拟這樣的過程,想到用一個vector存儲已經選出來的數,
再用map計數下即可.
*/
int n,x[maxn*maxn];
vector<int> ans;
map<int,int> mp;
map<int,int>::iterator it;
int main(){
    cin>>n;
    rep(i,0,n*n){
        cin>>x[i];
        mp[x[i]]++;
    }
    sort(x,x+n*n);
    for(int i=n*n-1;i>=0;i--){
        if(mp[x[i]]==0) continue;
        mp[x[i]]--;
        for(int j=0;j<ans.size();j++){
            int tmp=gcd(ans[j],x[i]);
            mp[tmp]-=2;
        }
        ans.push_back(x[i]);
    }
    for(int i=0;i<ans.size();i++) cout<<ans[i]<<" ";
    return 0;
}