4274: 约素

4274: 约素

Time Limit: 1 Sec  Memory Limit: 128 MB
Submit: 1802  Solved: 465

Description

判断一个正整数n的约数个数是否为p，其中p是素数。

Input

第一行给测试总数T(T <= 10000)。

接下来有T行，每行有两个数字n(1 <= n <= 1000000000)和p(2 < p <= 1000000000)。

Output

每组测试数据输出一行，如果n的约数个数是p，输出“YES”，否则输出“NO”。

Sample Input

564 7911 2331080 131024 1120170211 1913

Sample Output

YESNONOYESNO

HINT

Source

#include<cstdio>
#include<cstdlib>
#include<cstring>
using namespace std;
int count(int n)
{
    int ans=0;
    for(int i=1;i*i<=n;++i)
    {
        if(n%i==0)ans+=2;
        if(i*i==n)ans-=1;
    }
    return ans;
}
int main()
{
    int n,p,t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&p);
        if(count(n)==p)
        printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}