nyoj 813 对决

对决

时间限制：1000 ms  |           内存限制：65535 KB

难度：0

描述

Topcoder 招进来了 n 个新同学，Yougth计划把这个n个同学分成两组，要求每组中每个人必须跟另一组中每个同学进行一次算法对决，问存不存在一种分组方式在k场完成对决。（两组中每一组中人数都要大于0）

输入

有多组测试数据，每组测试数据两个数 n 和 k ，n和k都为0时表示输入结束。（0<n<10000,0<k<1000000）

输出

输出一行，如果可以，输出YES，不行的话输出NO。

样例输入

4 1
4 3
4 4
2 1
3 3
0 0

样例输出

NO
YES
YES
YES
NO

提示

4个人分成两组，1和3则需对决3场，2和2则需对决4场。

 

边看算法，边做以前剩下的水题还是有好处的，要不然会看晕的。

本来比较水的题不贴的，但是我错了一次才过，要长记性。

#include <cstdio>
#include <cmath>
using namespace std;
int main()
{
    int n, k;
    int i, j;
    int exist;
    while(scanf("%d%d", &n, &k), n||k)
    {
        if(n == 1)
        {
            printf("NO\n");
            continue;
        }
        exist = 0;
        for(i = 1;i <= n/2; i++)
        {
            if(i * (n-i) == k)
            {
                exist = 1;
                break;
            }
        }
        if(exist)
        printf("YES\n");
        else
        printf("NO\n");
    }
    return 0;
}

最优程序：

#include<stdio.h>
#include<math.h>
int main()
{
	int n,k,a,b;
	while(scanf("%d%d",&n,&k),n||k)
	{
		a=n*n-(k<<2),b=sqrt(a);
		puts(b*b!=a||(n+b)&1?"NO":"YES");
	}
}