Goffi and Squary Partition

Problem Description

Recently, Goffi is interested in squary partition of integers.

A set X of k distinct positive integers is called squary partition of n if and only if it satisfies the following conditions:

the sum of k positive integers is equal to n
one of the subsets of X containing k?1 numbers sums up to a square of integer.


For example, a set {1, 5, 6, 10} is a squary partition of 22 because 1 + 5 + 6 + 10 = 22 and 1 + 5 + 10 = 16 = 4 × 4.

Goffi wants to know, for some integers n and k, whether there exists a squary partition of n to k distinct positive integers.

Input

Input contains multiple test cases (less than 10000). For each test case, there‘s one line containing two integers n and k (2≤n≤200000,2≤k≤30).

Output

For each case, if there exists a squary partition of n to k distinct positive integers, output "YES" in a line. Otherwise, output "NO".

Sample Input

2 2
4 2
22 4

Sample Output

NO
YES
YES

题意为：

给出n,k，问能不能把n拆成k个不同的正整数相加，且其中k-1个数的和为完全平方数。

比如 n=22 ,k =4， 22可以拆成 1+5+6+10,且 1+5+10=4*4

BESTCODER主页上的题解：

PS：这道题目原来是要求输出一种可行方案的，所以下面题解是按照输出方案的思想搞的。
分析：
我们尝试枚举那个完全平方数 S,然后看能否将他拆分为 K-1 个数,并且不用到N-S
这一步可以用贪心+一次调整来搞定。为了保证 K-1 个数都不同,我们尝试尽量用 1,2,3...这些连续自然数来构造,如果 N-S 出现在这些数中,那么将 N-S 移除,再新加一个数。如果这样都不能拆分成 K-1 个数,那么这个 S 肯定不行。
现在考虑已经用上述方法拆分了,我们需要判断这个拆分是否可行。会产生问题的只有最后一个数,这个数可能和 N-S 一样,也可能出现在之前的序列。如果是出现在之前的序列,那么这个拆分也是不靠谱的。如果和 N-S 一样,那么分两种情况
1.	N-S 曾出现在之前的序列,那么显然这个拆分也是不靠谱的
2.	N-S 比倒数第二个数大,对于这种我们可以通过调整最后一个数和倒数第二个数的大小,来使得这个拆分成立,假设最后一个数为 a,倒数第二个为 b,只要 a-1,b+1 就好了。当然如果 a-1 = b+1 这个拆分也是不靠谱的这道题目就这样搞定了,其实没必要找所有的完全平方数,只要找小于 N 与 N 最接近的完全平方数就好了。

枚举出一个完全平方数，那么剩下的数就确定了，下面就开始判断剩下的那个数是否符合题意。

构造出完全平方数的k-1个数，从小到大，贪心，1,2,3.....

#include <iostream>
#include <cmath>
#include <stdio.h>
using namespace std;
int n,k;

bool judge(int s)//s为枚举出来的完全平方数
{
    int f=n-s;//剩下的
    int t=1;//t为构造出来的数
    for(int i=1;i<=k-2;i++)//开始构造k-1个数中的前k-2个，剩下的那个数为s
    {
        if(t==f)
            t++;
        s-=t;
        if(s<=t)
            return false;
        t++;
    }
    t--;
    //到此为止，t为构造出的k-1个数中的倒数第二个，s为倒数第一个
    while(1)
    {
        if(s<=t)//最后一个数出现在前面的序列中
            return false;
        if(s!=f)
            return true;
        //s==f时，调整s,t
        s--;
        t++;
        //s+t和不变。
    }
}

bool ok()
{
    int m=(int)sqrt(n);
    if(m*m==n)//枚举时要求完全平方数<n才可以
        m--;
    while(m)
    {
        if(judge(m*m))
            return true;
        m--;
    }
    return false;
}

int main()
{
    while(scanf("%d%d",&n,&k)==2)
    {
        if(ok())
            printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}