2019秋季PAT乙级真题--7-3 缘分数 (20 分)

7-3 缘分数 (20 分)

所谓缘分数是指这样一对正整数 a 和 b，其中 a 和它的小弟 a−1 的立方差正好是另一个整数 c 的平方，而 c 正好是 b 和它的小弟 b−1 的平方和。例如 83−73=169=132，而 13=32+22，于是 8 和 3 就是一对缘分数。

给定 a 所在的区间 [m,n]，是否存在缘分数？

输入格式：

输入给出区间的两个端点 0<m<n≤25000，其间以空格分隔。

输出格式：

按照 a 从小到大的顺序，每行输出一对缘分数，数字间以空格分隔。如果无解，则输出 No Solution。

输入样例 1：

8 200

输出样例 1：

8 3

105 10

输入样例 2：

9 100

输出样例 2：

No Solution

本题考散列优化算法，如果暴力，那么对每一个n，m区间内的数都需要计算不同的b的值，这样使得复杂度到了O（n^{2）的程度，必然是会超时的，但是找规律发现，对于每一个n，m来说，都需要遍历b的平方值，又因为对于遍历b来说，从小到大没一个数的平方和一定是递增的，所以，可以先用散列表把符合要求范围内的平方数存起来，题目限制范围最大到25000，计算sqrt（25000}3 - 24999^3 ）的值为43300.4，也就是说有情数b的平方和一定小于等于43300.4，这是一个范围上限，只需要使用一个arr数组，对i从小到大遍历，如果当前i的平方和超过了45000，那么直接结束循环（因为这里满足i递增的时候，平方和的值也递增，所以只要第一个i的平方超过了45000，就可以结束循环，即今后的每一个i都超过了45000）此时arr的下标对应的值中存的就是当前下标对应的原始平方和数，至此，只需要输入mn，遍历mn，如果符合题意即可。
注意：这里存在一个问题，sqrt函数输出值是一个double，要和arr的下标匹配，必须强制转换为int，但是这就出现了精度损失的问题，导致很多脏数据也被显示出来，为了解决这个问题，让强制转型后的sqrt和没有转型的sqrt相减，并取绝对值，如果该值小于0.001，那么可以判断符合要求

#include<iostream>
#include<cmath>
using namespace std;
const int maxn = 1e5+10;
int main() {
    int arr[maxn] = {0};
    for (int i = 1; i < 10000; ++i) {
        int temp = i * i + (i - 1) * (i - 1);
        if (temp > 50000)break;
        arr[temp] = i;
    }
    int m, n;
    cin >> m >> n;
    int key = 0;
    for (int i = m; i <= n; ++i) {
        int temp = (int) sqrt(i * i * i - (i - 1) * (i - 1) * (i - 1));
        if (arr[temp] != 0 && fabs(temp - sqrt(i * i * i - (i - 1) * (i - 1) * (i - 1))) < 0.001) {
            cout << i << ' ' << arr[temp] << endl;
            key = 1;
        }
    }
    if (!key)
        cout << "No Solution" << endl;
}