4274: 约素
Time Limit: 1 Sec Memory Limit: 128 MBSubmit: 1897 Solved: 496
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 <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#include <vector>
#include <set>
#include <bitset>
#include <stack>
#include <map>
#include <climits>
#include <functional>
using namespace std;
#define LL long long
const int INF=0x3f3f3f3f;
int main()
{
int t,n,p;
scanf("%d",&t);
while(t--)
{
scanf("%d %d",&n,&p);
int k=sqrt(n);
if((k*k)!=n) {printf("NO\n");continue;}
int sum=0;
for(int i=1;i<=k;i++)
{
if(n%i==0&&i==k) sum++;
else if(n%i==0) sum+=2;
}
if(sum==p) printf("YES\n");
else printf("NO\n");
}
return 0;
}