四平方和（解析）

题目描述

四平方和定理，又称为拉格朗日定理：

每个正整数都可以表示为至多 4 个正整数的平方和。

如果把 0 包括进去，就正好可以表示为 4 个数的平方和。

比如：

5 = 0^2 + 0^2 + 1^2 + 2^25=02+02+12+22；

7 = 1^2 + 1^2 + 1^2 + 2^27=12+12+12+22；

对于一个给定的正整数，可能存在多种平方和的表示法。

要求你对 4 个数排序：

0 \leq a \leq b \leq c \leq d0≤a≤b≤c≤d

并对所有的可能表示法按 a,b,c,da,b,c,d 为联合主键升序排列，最后输出第一个表示法。

输入描述

程序输入为一个正整数 N (N<5 \times 10^6)N(N<5×106)。

输出描述

要求输出 4 个非负整数，按从小到大排序，中间用空格分开

输入输出样例

示例

输入

12

输出

0 2 2 2

运行限制

• 最大运行时间：3s

• 最大运行内存: 256M

前面直接暴力，果然最大的数据会超时

#include <bits/stdc++.h>
using namespace std;
int main()
{
  int n;
  scanf("%d",&n);
  for(int i=0;i<=sqrt(n);i++) 
  for(int j=i;j<=sqrt(n);j++)
  for(int k=j;k<=sqrt(n);k++)
  for(int l=k;l<=sqrt(n);l++)//条件要写==，本身也是考虑在内的
  if(i*i+j*j+k*k+l*l==n)
  {
     printf("%d %d %d %d",i,j,k,l);
     return 0;

  }

  //每次接着前一次的数字是因为保证题目条件0≤a≤b≤c≤d
  //开方sqrt是因为最理想的情况就是0+0+0+sqrt(n)*sqrt(n);
  //后面以此类推，至于开方结果不为整数就不算，因为int 不保留小数
 
  return 0;
} 
//果然，还是有一个爆了，暴力算法的超时还真是轻松的
//所以说for循环真的慎用,或者说，有什么能够提高for循环的检索效率
//比如，加入if条件

AC

优化循环条件，优化遍历次数

#include <bits/stdc++.h>
using namespace std;
int main()
{
  int n;
  scanf("%d",&n);
  for(int i=0;i*i<=n;i++) 
  for(int j=i;i*i+j*j<=n;j++)
  for(int k=j;i*i+j*j+k*k<=n;k++)
  for(int l=k;i*i+j*j+k*k+l*l<=n;l++)//条件要写==，本身也是考虑在内的
  if(i*i+j*j+k*k+l*l==n)
  {
     printf("%d %d %d %d",i,j,k,l);
     return 0;

  }

  //程序不是提前预设时间复杂度的，而是实打实的根据结果得出的
//优化循环的条件,将循环次数降低，因为如果每次都遍历到最大，很可能比如
//i*i+j*j+k*k+l*l早已大过n了还在那里傻傻的遍历，等到这层循环结束.而如果控制每次循环一旦比n大
//我就退出循环就规避了无效遍历。
//而前面每次都少一个意思就是判断只有前面的数之和没有超过n我才需要进入下一个循环，
//如果超过了就需要更新前面的新的数字，也同样是避免了大量的无效循环，
//因此循环条件的设置也是能大大优化for的时间复杂度的.
 
  return 0;
} 
//优化循环条件，优化遍历次数