CF 583C GCD Table 思维

题意:定义:一个长度为n数列a,其gcd表为一个n*n的矩阵b, b[i][j]=gcd(a[i],a[j]) 
给出n*n个数,选出n个数作为数列a 使得数列a的gcd表正好为这n*n个数.
n<=500.数据保证一定有解.


gcd(a,b) <= min(a,b) 所以将table排序后.则前两大的数为a[m],a[m-1].结果存在res中.

因为一定有解.所以现在顺着对角线来填表,把已知的gcd加到map中.

若table中下一个数在map中 则该数为某两个数的gcd.否则该数为当前最大.填入对角线格子中(res)即可.O(n^2logn).

#include <bits/stdc++.h>
using namespace std;
const int N=1e6+5;
int n,a[N];
map<int,int> mp; 
int gcd(int a,int b)
{
	return b==0?a:gcd(b,a%b);
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cin>>n;
	for(int i=1;i<=n*n;i++)
		cin>>a[i];
	sort(a+1,a+1+n*n);
	reverse(a+1,a+1+n*n);
	vector<int> res;
	for(int i=1;i<=n*n;i++)	
	{
		if(mp[a[i]]==0)
		{
			for(int j=0;j<res.size();j++)
				mp[gcd(res[j],a[i])]+=2;
			if(res.size()<n)
				res.push_back(a[i]);
		}
		else
			mp[a[i]]--;
	}
	for(int i=0;i<res.size();i++)
		cout<<res[i]<<' ';
	return 0;
}