四平方和
暴力做法
Y总暴力做法,蓝桥云里能通过所有数据
总结:暴力也分好坏,下面这份代码就是写的好的暴力
如何写好暴力:1. 按组合枚举 2. 写好循环结束条件,没必要循环那么多次
#include<iostream>
#include<cmath>using namespace std;int n;int main(){std::ios::sync_with_stdio(false);cin.tie(0);cin>>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++){int t=(n-a*a-b*b-c*c);int d=sqrt(t);if(d*d==t){cout<<a<<" "<<b<<" "<<c<<" "<<d<<" ";return 0;}}return 0;
}
我自己写的暴力做法,只能过87.5% o(╥﹏╥)o
#include<iostream>
#include<cmath>using namespace std;int n;int main(){std::ios::sync_with_stdio(false);cin.tie(0);cin>>n;for(int a=0;a*a<=n;a++)for(int b=0;b*b<=n;b++)for(int c=0;c*c<=n;c++)for(int d=0;d*d<=n;d++){if(a*a+b*b+c*c+d*d==n){cout<<a<<" "<<b<<" "<<c<<" "<<d;return 0;}}return 0;
}
分析
参考题解
这道题目最重要的思路是空间换时间(这是算法竞赛里一个非常重要的思想!!!)
本来,我们需要枚举 a b c d 四个数字,因为第四个数字可以计算出来,所以至少需要三重循环,即 O(n3),但是这肯定是要超时的,所以就想办法优化循环的次数
- 重点来了:
一个比较好的思路是把三重循环拆成两次二重循环。在第一次二重循环中,计算一下c^2+d^2,然后记录下来,在第二次对a和b的循环中可以直接使用,而不需要再次计算。如此一来,时间复杂度就被大大的简化了。
至于如何记录 c^2+d^2 ,则可以考虑使用哈希表,或者数组+二分的做法。不得不说,实在是太巧妙了!!
二分做法
//四平方和
//二分
#include<bits/stdc++.h>using namespace std;const int N = 5e6 + 10;struct Sum{int s, c, d;// 下面这个重载用来解决按字典序输出的问题// 首先根据 c*c+d*d排序,为什么按c*c+d*d排序呢?// 因为我们的c是从小到大枚举,d从c开始枚举;也就是说我们的c和d是按字典序枚举的// 所以c*c+d*d中的c和d一定是符合字典序的// 如果c*c+d*d相同,那么就比较c;如果c*c+d*d和c都相同,那么就比较d;原因应该好理解bool operator<(const Sum &t)const{if(s != t.s) return s < t.s;if(c != t.c) return c < t.c;return d < t.d;}
}record[N * 2];int n;int main()
{cin >> n;int k = 0;// 用来记录结构体数组里的元素个数(也就是说record里有多少个元素)// 预处理(其实就是打表)for(int c = 0; c * c <= n; c++){for(int d = c; c * c + d * d <= n; d ++){record[k++] = {c * c + d * d, c, d};}}// 排序sort(record, record + k);for(int a = 0; a * a <= n; a ++){for(int b = a; a * a + b * b <= n; b++){int t = n - a * a - b * b;// k表示record里的元素个数int l = 0, r = k - 1;while(l < r){int mid = l + r >> 1;if(record[mid].s >= t) r = mid;else l = mid + 1;}if(record[l].s == t){printf("%d %d %d %d\n", a, b, record[l].c, record[l].d);return 0;}}}return 0;
}
哈希做法
使用C++自带的unordered_map过不了,不过模拟哈希表却可以过。
#include<iostream>
#include<cstring>
#include<algorithm>using namespace std;// 开放地址法的N一般为原先的两倍,在本题应该为2*5e6
// 但这里我怕爆内存,只好跟原来一样了;
// 注意N得为质数,5e6的质数为5e6+11
/* 求质数的代码:for(int i=5e6;;i++){bool flag=true;for(int j=2;j*j<=i;j++){if(i%j==0){flag=false;break;}if(flag){cout<<i<<endl;break;}}}*/
const int N=5e6+11,null=0x3f3f3f3f;int h[N];typedef pair<int,int> PII;PII s[N];int n;int find(int x){int k=(x%N+N)%N;while(h[k]!=null&&h[k]!=x){k++;}return k;}int main(){cin>>n;memset(h,null,sizeof h);// 打表for(int c=0;c*c<=n;c++)for(int d=c;c*c+d*d<=n;d++){int k=find(c*c+d*d);if(h[k]==null) {h[k]=c*c+d*d;s[k].first=c,s[k].second=d;}}for(int a=0;a*a<=n;a++)for(int b=a;a*a+b*b<=n;b++){int t=n-a*a-b*b;int k=find(t);if(h[k]!=null){cout<<a<<" "<<b<<" "<<s[k].first<<" "<<s[k].second;return 0;}}return 0;
}