【蓝桥杯】2016初赛 四平方和 (暴力优化）

题目描述

四平方和定理，又称为拉格朗日定理：每个正整数都可以表示为至多4个正整数的平方和。
如果把0包括进去，就正好可以表示为4个数的平方和。
比如：
5 = 0^2 + 0^2 + 1^2 + 2^2
7 = 1^2 + 1^2 + 1^2 + 2^2（符号表示乘方的意思）
对于一个给定的正整数N，可能存在多种平方和的表示法。
要求你对4个数排序：0 <= a <= b <= c <= d
并对所有的可能表示法按 a,b,c,d 为联合主键升序排列，最后输出第一个表示法

输入

输入存在多组测试数据，每组测试数据输入一行为一个正整数N (N<5000000)

输出

对于每组测试数据，要求输出4个非负整数，按从小到大排序，中间用空格分开

样例

** 样例输入**

5
12
773535

样例输出

0 0 1 2
0 2 2 2
1 1 267 838

【想说的】：不是一个很难得题目，却很玄学，两重循环加哈希表，居然超时，而四重循环改三重循环却过了。讲道理分析一下时间复杂度得话三重循环应该会爆掉的啊。所以记录一下，可太暴力了

题解

双重循环加哈希表

#include <bits/stdc++.h>
#define ll long long
using namespace std;
map<int,int> hashh;
int main(){
    int n;
    scanf("%d",&n);

        for(int c=0;c*c<=n/2;c++)
        {
            for(int d=c;c*c+d*d<=n;d++)
            {
                if(hashh.find(c*c+d*d)==hashh.end())
                hashh[c*c+d*d]=c;
            }
        }
        for(int a=0;a*a<=n/4;a++)
        {
            for(int b=a;a*a+b*b<=n/2;b++)
            {
                if(hashh.find(n-a*a-b*b)!=hashh.end())
                {
                    int c=hashh[n-a*a-b*b];
                    int d=int(sqrt(n-a*a-b*b-c*c));
                    printf("%d %d %d %d\n",a,b,c,d);
                    return 0;
                }
            }
        }


    return 0;
}

三重循环

#include <bits/stdc++.h>
#define ll long long
using namespace std;
int main()
{
	int n;
	scanf("%d",&n);
	for(int a=0;a*a<n;a++)
	{
		for(int b=a;a*a+b*b<n;b++)
		{
			for(int c=b;a*a+b*b+c*c<n;c++)
			{
				if(pow(int(sqrt(n-a*a-b*b-c*c)),2)==n-a*a-b*b-c*c)
				{
					printf("%d %d %d %d\n",a,b,c,int(sqrt(n-a*a-b*b-c*c)));
					return 0;
				}
				
			}
		}
	}
}