poj2100尺取法

题意:

分解整数,将一个整数分解成多个数的平方和;

理解:

开始我以为可以暴力枚举,然后发现n是1e14次方;

过不了,然后翻书发现是尺取法;

就用尺取法写了;

但是报了我一个超空间...

随后算了下空间...果然用了1e7个longlong的字节;

然后想啊....

最后受大神影响,发现可以不用数组...

结果只用了180k...

代码如下:

#include <cstdio>

#include <algorithm>
#include <vector>

using namespace std;

typedef long long ll;
typedef pair<int, int> pii;

#define x first
#define y second

ll n;
vector<pii> v_a;

void search()
{
    ll sum = 0, j = 1, i = 1;
    while (i * i <= n && j * j <= n)
    {
        while (sum < n && i * i <= n)
        {
            sum += i * i;
            ++i;
        }
        while (sum > n)
        {
            sum -= j * j;
            ++j;
        }
        if (sum == n)
        {
            v_a.push_back(pii(j, i - j));
            sum -= j * j;
            ++j;
        }
    }
}

bool cmp(pii p1, pii p2)
{
    return p1.x > p2.x;
}

int main()
{
    scanf("%lld", &n);

    search();

    printf("%d\n", v_a.size());
    sort(v_a.begin(), v_a.end(), cmp);
    for (int i = v_a.size() - 1; i >= 0; --i)
    {
        printf("%d ", v_a[i].y);
        for (int j = v_a[i].x; j < v_a[i].x + v_a[i].y; ++j)
            printf("%d ", j);
        printf("\n");
    }

    return 0;
}