输入两个正整数a ,b ，请判断a+b和a*b是否为质数，是就输出“YES”，否就输出“NO”。

#include<bits/stdc++.h>
using namespace std;
int a,b;
int main(){
    scanf("%d%d",&a,&b);
    int flag=0;
    for(int i=2;i<a+b;i++){
        if((a+b)%i==0){
            flag=1;
            break;
        }
    }
    if(flag==0) printf("YES\n");
    else printf("NO\n");
    flag=0;
    for(int i=2;i<a*b;i++){
        if((a*b)%i==0){
            flag=1;
            break;
        }
    }
    if(flag==0) printf("YES");
    else printf("NO");
    return 0;
}