CS 400 Coprime Pairs 构造!!!

题意:给出n和k 构造出n个不同的数,任意选两个有C(n,2)对,满足正好有k对是互质的. n<=1e4,0<=k<=C(n,2). a[i]<=1e6.


考虑简单的情况 当k=0 取序列为前n个素数即可.
转换一下问题 求不互质的有y=C(n,2)-k对时的序列.
构造:另序列为5,7,11,13...  现在任取一对都为互质.
找到最大的T 使得C(T,2)<=k 将a[1],a[2]...a[T]都乘上2 

现在还需要x=k-C(T,2)对,将前x个乘上3,a[T+1]*3即可.

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e6+5;
ll n,k;
int vis[N];
vector<int> p;
void init()
{
	for(int i=2;i<N;i++)
		if(!vis[i])
			for(int j=2*i;j<N;j+=i)
				vis[j]=1;
	for(int i=5;i<N;i++)
		if(!vis[i])
			p.push_back(i);
}
int a[N];
int main()
{
	init();
	cin>>n>>k;
	for(int i=0;i<n;i++)
		a[i]=p[i];
	k=n*(n-1)/2-k;
	ll T=1;
	while(T*(T-1)/2<=k)
		T++;
	T--;
	for(int i=0;i<T;i++)
		a[i]*=2;
	k-=T*(T-1)/2;
	for(int i=0;i<k;i++)
		a[i]*=3;
	a[T]*=3;
	for(int i=0;i<n;i++)
		printf("%d ",a[i]);	
	return 0;
}