以下C++代码判断输入的正整数是否为质数，如果该数字是质数，则输出 YES ，否则输出 NO 。请在横线上填写代码。

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