【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。

 1 #include <cstdio>
 2 #include <cmath>
 3 
 4 int n, k;
 5 
 6 bool judge(int s) {
 7     int p, q;
 8     int i, j = 0, sum = 0, tmp;
 9 
10     p = s*s;
11     if (n == p)
12         return false;
13     q = n-p;
14 
15     for (i=1; i<=k-2; ++i) {
16         ++j;
17         if (j == q)
18             ++j;
19         sum += j;
20     }
21     if (sum >= p)
22         return false;
23     tmp = p - sum;
24     if (tmp <= j)
25         return false;
26     if (tmp == q) {
27         if (q <= j+2)
28             return false;
29     }
30     return true;
31 }
32 
33 int main() {
34     int m;
35 
36     while (scanf("%d %d", &n, &k) != EOF) {
37         m = sqrt((n-1)*1.0);
38 
39         if (judge(m))
40             printf("YES\n");
41         else
42             printf("NO\n");
43     }
44 
45     return 0;
46 }

 

posted on 2014-08-27 18:25  Bombe  阅读(160)  评论(0编辑  收藏  举报

导航