【蓝桥杯】试题 历届真题 四平方和【第七届】【省赛】【A组】

纯暴力，也就是四个for循环过不了。拿哈希表优化一下，这让我想起了leetcode第一题两数之和用哈希表优化到O(N)，map<int,int>第一个int是第二个int的平方，这样只需要三个for循环就行了，最后一个数用哈希表找。

#include<iostream>
#include<vector>
#include<algorithm>
#include<map>
#include<math.h>
#include<set>
#include<string>
#include<stack>
using namespace std;
#include<queue>
#include<sstream>


int main()
{
	map<int, int>map1;
//哈希表预处理，提前存下所有可能的平方
	for (int i = 0; i <= 2237; i++)
	{
		map1[i * i] = i;
	}
	int n = 0;
	cin >> n;
	for (int a = 0; a <= 2237; a++)
	{
		if (a * a > n)
		{
			break;
		}
		for (int b = 0; b <= 2237; b++)
		{
			if (a * a + b * b > n)
			{
				break;
			}
			for (int c = 0; c <= 2237; c++)
			{
				if (a * a + b * b + c * c > n)
				{
					break;
				}
				if (map1.find((n - a * a - b * b - c * c)) != map1.end())
				{
					cout << a << " " << b << " " << c << " " << map1[(n - a * a - b * b - c * c)];
					return 0;
				}
				
			}
		}
	}
	
	return 0;
}