以下C++代码判断输入的正整数是否为质数，如果该数字是质数，则输出 YES，否则输出 NO。质数是指仅能被1和它本身整除的正整数。请在横线上填写代码。（ ）

int num, i;
cin >> num;
for(i = 2; i < num; i++)
    if (__________){
        cout << ("NO");
        break;
    }
if(i == num)
    cout << ("YES");