Luogu P5316 【恋恋的数学题】

是个神仙题

就三种情况，分类讨论。

(k=2)：

因为保证有解，所以直接输出即可。

(k=3)：

由于对应情况可以枚举全排列寻找，所以在此只考虑顺序对应时的情况，不妨设六个数分别为(g_{ab},g_{ac},g_{bc},l_{ab},l_{ac},l_{bc})，由小学数学知识可知(g_{ab}l_{ab}=ab,g_{ac}l_{ac}=ac,g_{bc}l_{bc}=bc)，又(largefrac{abcdot ac}{bc} ormalsize=a^2)，所以(a=sqrt{largefrac{abcdot ac}{bc}})，同理(b=sqrt{largefrac{abcdot bc}{ac}},c=sqrt{largefrac{accdot bc}{ab}})。需要注意的是直接算(abcdot ac)会爆__int128，在此改为计算(Largefrac{ab}{gcd(ab,bc)}cdotfrac{ac}{frac{bc}{gcd(ab,bc)}})。

(k=4)：

同(k=3)，不妨设所有数分别为(g_{ab},g_{ac},g_{bc},g_{ad},g_{bd},g_{cd},l_{ab},l_{ac},l_{bc},l_{ad},l_{bd},l_{cd})，由于已知(k=3)的解法，分别求(a,b,c)，(a,b,d)即可，最后合并答案。

#include<bits/stdc++.h>

using namespace std;

typedef long long LL;
typedef __int128 LLL;

LLL gcd(LLL u,LLL v){
	if(v==0){
		return u;
	}
	return gcd(v,u%v);
}

LLL lcm(LLL u,LLL v){
	return u/gcd(u,v)*v;
}

LLL sqrt(LLL u){
	LLL mid,l=1,r=1e18;
	while(l<=r){
		mid=(l+r)>>1;
		if(mid*mid==u){
			return mid;
		}
		else if(mid*mid<u){
			l=mid+1;
		}
		else{
			r=mid-1;
		}
	}
	return 0;
}

LLL div(LLL a,LLL b,LLL c){
	LLL _g=gcd(a,c);
	c/=_g;
	return (a/_g)*(b/c);
}

int t,k,n;

LL g[10],l[10],ans[10];

vector<LL>solve3(LLL g12,LLL g13,LLL g23,LLL l12,LLL l13,LLL l23){
	LLL ans1,ans2,ans3,x12,x13,x23;
	x12=g12*l12;
	x13=g13*l13;
	x23=g23*l23;
	ans1=sqrt(div(x12,x13,x23));
	ans2=sqrt(div(x12,x23,x13));
	ans3=sqrt(div(x23,x13,x12));
	if(ans1*ans2*ans3==0){
		return {};
	}
	if(gcd(ans1,ans2)==g12&&lcm(ans1,ans2)==l12&&
	gcd(ans2,ans3)==g23&&lcm(ans2,ans3)==l23&&
	gcd(ans1,ans3)==g13&&lcm(ans1,ans3)==l13){
		return {(LL)ans1,(LL)ans2,(LL)ans3};
	}
	else{
		return {};
	}
}

void solve(){
	if(k==2){
		ans[1]=g[1];ans[2]=l[1];
	}
	else if(k==3){
		for(int i=6;i;i--,next_permutation(g+1,g+4)){
			for(int j=6;j;j--,next_permutation(l+1,l+4)){
				auto rep=solve3(g[1],g[2],g[3],l[1],l[2],l[3]);
				if(rep.empty()){
					continue;
				}
				ans[1]=rep[0];ans[2]=rep[1];ans[3]=rep[2];
				return;
			}
		}
	}
	else{
		for(int i=720;i;i--,next_permutation(g+1,g+7)){
			for(int j=720;j;j--,next_permutation(l+1,l+7)){
				auto rep1=solve3(g[1],g[2],g[3],l[1],l[2],l[3]),rep2=solve3(g[1],g[4],g[5],l[1],l[4],l[5]);
				if(rep1.empty()||rep2.empty()){
					continue;
				}
				if(rep1[0]!=rep2[0]||rep1[1]!=rep2[1]){
					continue;
				}
				if(gcd(rep1[2],rep2[2])!=g[6]||lcm(rep1[2],rep2[2])!=l[6]){
					continue;
				}
				ans[1]=rep1[0];ans[2]=rep1[1];ans[3]=rep1[2];ans[4]=rep2[2];
				return;
			}
		}
	}
}

int main(){
	scanf("%d%d",&t,&k);
	n=(k-1)*k/2;
	while(t--){
		for(int i=1;i<=n;i++){
			scanf("%lld",&g[i]);
		}
		for(int i=1;i<=n;i++){
			scanf("%lld",&l[i]);
		}
		solve();
		for(int i=1;i<=k;i++){
			printf("%lld%c",ans[i]," 
"[i==k]);
		}
	}
	return 0;
}