【蓝桥杯】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

py直接暴力会超时

先枚举a和b，注意范围的优化
反向带入求c，符合条件输出

import math
while 1:
    n = int(input())
    t=1
    for a in range(1,int(math.sqrt(n))+1):
        for b in range(a,int(math.sqrt(n-a*a))+1):
            c = n-a**2-b**2
            c = math.sqrt(c)
            if c-int(c)==0 and c>=b:
                print(a,b,int(c))
                t=0
    if t:
        print("No Solution")