素數判定（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;
}