題目描述

給出多個整數n,要求求出a,b,c字典序最小的 a2+b2+c2=n 

輸入

一個整數t,表明資料組數 (t <= 10) 
接下來t行,每行一個整數n (1 <= n <= 10000000) 

輸出

三個整數 a,b,c 
如果無解,輸出"???" 

樣例輸入

2
7
30

樣例輸出

???
1 2 5

來源/分類

2018新生賽 八

相比於前一個，複雜度增加，應採取措施。

用一個vis陣列來判斷並存儲。

#include<cstdio>
const int maxn=1e7+5;
int vis[maxn];
int main()
{
	//freopen("in.txt","r",stdin);
	int t;
	scanf("%d",&t);
	for(int i=1;i*i<=10000000;i++) vis[i*i]=i;
	
	while(t--){
		int n,flag=0;
		scanf("%d",&n);
		int i=0,j=0;
		for(i=0;i*i<=n;i++){
			for(j=i;j*j<=n;j++){
					if(n-i*i-j*j>=1&&vis[n-i*i-j*j]){//n-i*i-j*j是平方數 
						flag=vis[n-i*i-j*j];
						break;
					}
				}
				if(flag) break;
		}
		if(flag) printf("%d %d %d\n",i,j,flag);
		else printf("???\n");
	}
}