第5关：循环结构：是素数吗

编程要求

在右侧编辑器中的Begin-End之间补充代码，判断输入的正整数（数据由平台提供，你需要获取后使用）是否为素数，如果是，则输出 Yes，否则输出 No。

注意：程序保证输入的正确性，即输入的 n 为正整数。

提示

本关需要用到循环控制结构来解决。如果输入的整数保存在变量 k 中，则需要尝试一些数能否被 k 整除，并根据结果判断 k 是否是素数。需要测试的数据可以是从 2 到k-1，或者从 2 到k/2，或者从 2 到开根号 k 也可以。

测试说明

平台会对你编写的代码进行测试，比对你输出的数值与实际正确数值，只有所有数据全部计算正确才能通过测试：

测试输入：2
预期输出：Yes

测试输入：987
预期输出：No

// 包含两种I/O库，可以使用任一种输入输出方式
#include <stdio.h>
#include <iostream>

using namespace std;

bool isPrime(int num) {
    for (int i = 2; i * i <= num; ++i) {
        if (num % i == 0) {
            return false;
        }
    }
    return true;
}

int main() {
    int n;
    // 请在此添加代码，输入正整数n，如果n是素数则输出“Yes”，否则输出“No”
    /********** Begin *********/
    cin >> n;
    bool flag = isPrime(n);
    if (flag) {
        cout << "Yes";
    } else {
        cout << "No";
    }
    /********** End **********/
    return 0;
}