51nod 1080 两个数的平方和

1080 两个数的平方和 

基准时间限制：1 秒 空间限制：131072 KB 分值: 5  难度：1级算法题

 收藏

 关注

给出一个整数N，将N表示为2个整数i j的平方和（i <= j)，如果有多种表示，按照i的递增序输出。

例如：N = 130，130 = 3^2 + 11^2 = 7^2 + 9^2 （注：3 11同11 3算1种）

Input

一个数N(1 <= N <= 10^9)

Output

共K行：每行2个数，i j，表示N = i^2 + j^2（0 <= i <= j)。
如果无法分解为2个数的平方和，则输出No Solution

Input示例

130

Output示例

3 11
7 9

思路：预处理小于n的所有平方数。然后二分,(nlogn)

说下两个常用的STL二分查找方法 .

lower_bound(a, a + n, k) - a;  前两个参数是查找范围，左闭右开。第三个参数是值， 返回的是地址，所以 -a 可以得到下标。

 在数组[0,n)中找到第一个大于等于k的下标，找不到返回n，即最后一个元素的下一个位置.

同理 upper_bound(a, a + n, k) -a;

  在数组[0,n)中找到第一个大于k的下标，找不到返回n）

code ：

#include <bits/stdc++.h>
using namespace std;
const int MAX = (int)1e9;
set<pair<int,int> > iset;
int a[50000] = {0};
int main(void) {
    int n ;
    cin >>n;
    int j = -1;
    for(int i = 0; ; i++) {
        a[++j] = i * i;
        if(a[j] > n) break;
    }
      for (int i = 0; i < j; i++) {
        int tem = i * i;
         int k = lower_bound(a, a + j, n - tem)-a;
        if(k < j && a[k] == n - tem) {
            int x = i;
            int y = k;
            if(x > y ) swap(x,y);
            iset.insert(make_pair(x,y));
        }
    }
    set<pair<int,int> >::iterator it;
    if (!iset.empty()) {
        for (it = iset.begin(); it != iset.end() ;it++) {
            cout << it->first << " " << it->second << endl;
        }
    }
    else {
        cout << "No Solution" << endl;
    }
    return 0;
}