【HDOJ】4982 Goffi and Squary Partition

题意就是整数划分，选出和为n的K个整数，其中K-1个数的和为完全平方数S。
选择整数时需要从1,2,3..连续选择，当选择整数与n-S相等时，需要跳过n-S，即选择n-S+1。
如此选择K-2个数，从而可确定第K-1个数，若该数已经出现（小于或等于K-2），则划分失败；
若第K-1个数不等于n-S，则肯定划分成功，否则K-1个数若等于n-S。
即需要通过将第K-2个数+1，同时第K-1个数-1得到正确的划分，并且需要保证调整后第K-2个数仍小于第K-1个数，因此，两数之间的距离至少大于2。

#include <cstdio>
#include <cmath>

int n, k;

bool judge(int s) {
    int p, q;
    int i, j = 0, sum = 0, tmp;

    p = s*s;
    if (n == p)
        return false;
    q = n-p;

    for (i=1; i<=k-2; ++i) {
        ++j;
        if (j == q)
            ++j;
        sum += j;
    }
    if (sum >= p)
        return false;
    tmp = p - sum;
    if (tmp <= j)
        return false;
    if (tmp == q) {
        if (q <= j+2)
            return false;
    }
    return true;
}

int main() {
    int m;

    while (scanf("%d %d", &n, &k) != EOF) {
        m = sqrt((n-1)*1.0);

        if (judge(m))
            printf("YES
");
        else
            printf("NO
");
    }

    return 0;
}