acwing 1221. 四平方和（蓝桥杯）

题目：1221. 四平方和

思路：可知四个数里的最大数m的平方不可能超过n，即可对a,b,c,d从1~sqrt(n)遍历,先对c,d进行遍历，并将c * c+d * d的和存起来（对于c * c+d * d只存第一次出现的c和d），然后再对a,b进行遍历，去找之前存起来的数里有没有（n-a * a-b*b）即可

方法一：用的是map来存储c*c+ d * d,发现会超时

#include<bits/stdc++.h>

using namespace std;
typedef long long LL;
const int N=5e6+10;
typedef pair<int ,int > PII;
unordered_map<int,PII> mp;
int main(){
    int n;
    cin>>n;
    int len=sqrt(n)+1;
    for(int i=0;i<=len;i++){
        for(int j=i;j<=len;j++){
            if((i*i+j*j)>n) break;
            if(mp.find(i*i+j*j)==mp.end()){
                mp[i*i+j*j]={i,j};
            }
        }
    }
    bool flag=1;
    for(int i=0;i<=len;i++){
        for(int j=i;j<=len;j++){
            if((i*i+j*j)<=n&&mp.find(n-i*i-j*j)!=mp.end()){
                cout<<i<<" "<<j<<" ";
                cout<<mp[n-i*i-j*j].first<<" "<<mp[n-i*i-j*j].second;
                flag=0;
                break;
            }
        }
        if(flag==0) break;
    }
    return 0;
}

方法二：这次是用int a[N]来存，只存c，另外一个d可以通过计算得到

#include<bits/stdc++.h>

using namespace std;
typedef long long LL;
const int N=5e6+10;
typedef pair<int ,int > PII;
int a[N];
int main(){
    int n;
    cin>>n;
    int len=sqrt(n)+1;
    for(int i=0;i<=len;i++){
        for(int j=i;j<=len;j++){
            if((i*i+j*j)>n) break;
            if(!a[i*i+j*j]){
                a[i*i+j*j]=i+1;
            }
        }
    }
    bool flag=1;
    for(int i=0;i<=len;i++){
        for(int j=i;j<=len;j++){
            if((i*i+j*j)>n) break;
            if(a[n-i*i-j*j]){
                cout<<i<<" "<<j<<" ";
                cout<<a[n-i*i-j*j]-1<<" "<<sqrt(n-i*i-j*j-(a[n-i*i-j*j]-1)*(a[n-i*i-j*j]-1));
                flag=0;
                break;
            }
        }
        if(flag==0) break;
    }
    return 0;
}