【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 = , sum = , tmp;

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

     for (i=; i<=k-; ++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+)
             return false;
     }
     return true;
 }

 int main() {
     int m;

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

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

     return ;
 }