[蓝桥杯2015初赛]方程整数解

题目描述

方程: a^2 + b^2 + c^2 = 1000
这个方程有正整数解吗？有：a,b,c=6,8,30 就是一组解。
求出 a^2 + b^2 + c^2 = n(1<=n<=10000)的所有解，解要保证c>=b>=a>=1。

输入格式

存在多组测试数据，每组测试数据一行包含一个正整数n(1<=n<=10000)

输出格式

如果无解则输出"No Solution"。
如果存在多解，每组解输出1行，输出格式：a b c，以一个空格分隔
按照a从小到大的顺序输出，如果a相同则按照b从小到大的顺序输出，如果a,b都相同则按照c从小到大的顺序输出。

输入样例

4
1000

输出样例

No Solution
6 8 30
10 18 24

思路：模拟+暴力，需要优化一下时间复杂度，通过判定，去掉一些不可能解即可

#include<bits/stdc++.h>
using namespace std;
int m,upper,lower,flag;
int main(){
	while(scanf("%d",&m) != EOF){
		flag = 0;
		upper = sqrt(m);
		for(int i = 1; i <= upper; i++){
			for(int j = i; j <= upper; j++){
				for(int k = j; k <= upper; k++){
					if(i*i+j*j+k*k > m) break;
					if(i*i+j*j+k*k == m){
						flag = 1;
						cout << i << " " << j << " " << k << endl;
					}
				}
			}
		}
		if(flag == 0) cout << "No Solution" << endl;
	}
}