【二分法】四平方和

Nathan Qian

已于 2022-01-26 16:00:40 修改

阅读量585

点赞数

分类专栏：算法文章标签：蓝桥杯 c++ 二分法

于 2022-01-26 15:59:48 首次发布

本文链接：https://blog.csdn.net/nathanqian123/article/details/122703583

版权

算法专栏收录该内容

52 篇文章 0 订阅

订阅专栏

题目

解释

这道题有两个解法，一个是建立哈希表，一个是二分查找，这里采取二分查找法加暴力循环
要求找四个数a,b,c,d
首先对前两个数c,d进行循环打表，存入结构体sum中（空间换时间）
然后固定一个数a进行循环，另一个数b采用二分法查找，以此来降低原本为o( $n\sqrt{n}$ )的时间复杂度

代码段

#include<iostream>
#include<algorithm>
using namespace std;
const int N = 25e5 + 10;
struct s
{
    int s, c, d;
}sum[N];
bool cmp(s t1, s t2)
{
    if (t1.s != t2.s)return t1.s < t2.s;
    if (t1.c != t2.c)return t1.c < t2.c;
    if (t1.d != t2.d)return t1.d < t2.d;
}
int m;
int main()
{
    int n; cin >> n;
    for (int c = 0; c * c <= n; c++)
        for (int d = c; d * d + c * c <= n; d++)sum[m++] = { c * c + d * d,c,d };
//为防止重复,d必须从c开始循环
    sort(sum, sum + m, cmp);
    for (int a = 0; a*a  <= n; a++)
        for (int b = 0; b * b + a * a <= n; b++)
        {
            int t = n - a * a - b * b;
            int l = 0, r = m - 1;
            while (l < r)
            {
                int mid = (l + r) / 2;
                if (sum[mid].s >= t)
                    r = mid;
                else
                    l = mid + 1;
            }
            if (sum[l].s == t)
            {
                cout << a << ' ' << b << ' ' << sum[l].c << ' ' << sum[l].d << endl;
                return 0;
            }
        }
    return 0;

}