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 secondll 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;}