每日一题-四平方和（蓝桥杯）（暴力/哈希表）

暴力解法（会超时）

#include<bits/stdc++.h>
using namespace std;
int n;
unordered_map<int ,int>mp;
void init()
{
    for(int i=0;i*i<=n;i++)
    {
        mp[i*i]++;
    }
}

int main()
{
    cin>>n;
    init();
    int x = (int)sqrt(n);
    for(int i=0;i<=x;i++)
    {
        for(int j=i;j<=x;j++)
        {
            for(int k=j;k<=x;k++)
            {
                int xx = n-i*i-j*j-k*k;
                // cout<<xx<<'\n';
                if(xx<0) continue;
                if(mp.count(xx))
                {
                    cout<<i<<" "<<j<<" "<<k<<" "<<sqrt(xx)<<"\n";
                    return 0;
                }
            }
        }
    }
    
    return 0;
}

hash表（也超时了）
具体思路就是先把后边两项的平方和存进hash表，存入的值为字典序小的值。然后在对前两项枚举，如果算出来的后边两项的值存在于hash表中，计算答案。

#include<bits/stdc++.h>
using namespace std;
int n;
unordered_map<int ,int>mp;
void init()
{
    for(int i=0;i*i<=n;i++)
    {
        for(int j=i;i*i+j*j<=n;j++)
        {
            if(!mp.count(i*i+j*j))
                mp[i*i+j*j] = i;
        }
    }
}

int main()
{
    cin>>n;
    init();
    for(int i=0;i*i*4<=n;i++)
    {
        for(int j=i;j*j*2<=n;j++)
        {
            if(mp.count(n-i*i-j*j))
            {
                int c = mp[n-i*i-j*j];
                int d = sqrt(n-i*i-j*j-c*c);
                cout<<i<<" "<<j<<" "<<c<<" "<<d<<'\n';
                return 0;
                
            }
        }
    }
    return 0;
}

标签：hash,cout,int,平方和,蓝桥,init,mp,哈希,两项
来源： https://blog.csdn.net/qq_50285142/article/details/115079104