【华为OD笔试题】计算[a，b]范围内，互质的勾股数，并输出全部两两互质的勾股数

华为笔试题：计算互质的勾股数

给出输入：
1 20
输出：
3 4 5
5 12 13
8 15 17
输入输出的解释：
给出三个数在1-20范围内的勾股数，而且三个数必须两两互质（也就是最大公约数为1）
例如（3，4，5）（5，12，13）（8，15，17），
其中（6，8，10）因为6，8的最大公因子是2，所以不能达成互质条件，所以不能包含其中
而且注意，要保持按照a升序，b升序，c升序的顺序排列

如果输入为
4 5
输出：
NA

思路：

这个题其实我不是很清楚勾股数的快速解法，而且当时第一次做笔试超级紧张

我觉得考试这种压力大的时候，可以不要太准追求完美，
仔细思考哪里可以优化就可以了
我提交了三次，分别优化了

1. $a^2+b^2=c^2$，已知a,b,求c，这里之前用的是从a开始遍历到$a^2+b^2$，如果找到$c\ast c==a^2+b^2$，就返回；显然这样算复杂度很高，在想不出更简单的办法的情况下，我用了一个笨办法，sqrt(num) - 1和sqrt(num) + 1，这样最多只需要算3次，就能得到c
2. 考虑start>=end的特殊情况，直接返回，

最后虽然写的不是很优雅，但是也100% pass了

#include <iostream>
#include<vector>
#include<algorithm>
#include<cmath>
using namespace std;
/*
* a<b,求最大公约数
*/
int gcd(int a, int b) {
    
    
    return b == 0 ? a : gcd(b, a % b);
}

int getNumSqrt(int a, int b) {
    
    
	int num = pow(a, 2) + pow(b, 2);
	for (int i = sqrt(num)-1; i < sqrt(num)+1; i++)
	{
    
    
		if (num == (int)pow(i, 2)) {
    
    
			return i;
		}
	}
	return -1;
}


void GetTri(int start,int end) {
    
    
	if (start>=end)
	{
    
    
		cout << "NA" << endl;
		return;
	}
	vector<vector<int>> result;
	for (int i = start; i < end; i++)
	{
    
    
		for (int j = i+1; j < end; j++) {
    
    
			int c = getNumSqrt(i, j);
			
			if (c!=-1&&c<=end)
			{
    
    
				if (gcd(i, j)==1 && gcd(j, c)==1 && gcd(i, c)==1)
				{
    
    
					vector<int> abc = {
    
     i,j,c };
					result.push_back(abc);
				}
			}
		}
	}
	if (result.size()==0)
	{
    
    
		cout << "NA"<<endl;
		return;
	}
	for (int i = 0; i < result.size(); i++)
	{
    
    
		for (auto j : result[i]) {
    
    
			cout << j << " ";
		}
		cout << endl;
	}
}


int main()
{
    
    
	int s,e;
	cin >> s>>e;
	GetTri(s,e);
}