1221. 四平方和

shortBackboard

已于 2023-11-07 17:15:30 修改

阅读量46

点赞数 1

分类专栏： AcWing蓝桥杯C++辅导AB组文章标签： c++ 数据结构算法

于 2023-10-25 22:53:02 首次发布

本文链接：https://blog.csdn.net/m0_73781016/article/details/134044867

版权

AcWing蓝桥杯C++辅导AB组专栏收录该内容

30 篇文章 0 订阅

订阅专栏

1.暴力算法时间复杂度O(n^3)

2.哈希表，时间复杂度降为O(n^2)

/*
  这四个数都小于等于根号n
  
  模拟哈希表，记录c^2 + d^2 的值
*/

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>

using namespace std;

const int N = 5e6 + 10;

int n;
int C[N], D[N];

int main()
{
    cin >> n;
    
    // 都设置为-1
    memset(C, -1, sizeof C);
    
    // d从c开始，保证记录的值为升序
    for(int c = 0; c * c <= n; c++)
        for(int d = c; d * d + c * c <= n; d++)
        {
            int s = c * c + d * d;
            
            if(C[s] == -1)
            {
                C[s] = c;
                D[s] = d;
            }
        }
        
    for(int a = 0; a * a <= n; a++)
        for(int b = a; b * b + a * a <= n; b++)
        {
            int s = n - a * a - b * b;
            
            if(C[s] != -1)
            {
                printf("%d %d %d %d\n", a, b, C[s], D[s]);
                return 0;
            }
        }
    
    
    return 0;
}