素数判定（int sushu(int n))

题目描述
输入一个正整数n，判断n是否是素数，若n是素数，输出”Yes”,否则输出”No”。
输入
输入一个正整数n(n<=1000)
输出
如果n是素数输出"Yes"，否则输出"No"。输出占一行。
样例输入 Copy
2
样例输出 Copy
Yes

# include<stdio.h>
# include<math.h>
# pragma warning(disable : 4996)
int main()
{
    
    
    int n;
    scanf("%d", &n);
    int k,i;
    k = (int)sqrt(n);
    for (i = 2;i <= k; i++)
    {
    
    
        if (n % i == 0)
            break;
    }
    if (i<=k||n==1)
        printf("No\n");
    else
        printf("Yes\n");
    return 0;
}