[2553] - 完美的素数(C++版本)

#include <iostream>
using namespace std;
int haha(int x)      //判断素数
{
    int a=0;
    for(int i=2; i<x; i++)    //若a不等于0，则x非素数
    {
        if(x%i==0)
        {
            a++;
            break;
        }
    }
    return a;
}
int main()
{
    int m,sum=0,n=0;
    while(cin>>m)
    {
        n=m;
        for(int i=1; m!=0; i++)
        {
            sum=sum+m%10;
            m=m/10;
        }
        if(haha(n)==0&&haha(sum)==0)
            cout<<"YES"<<endl;
        else
            cout<<"NO"<<endl;
        sum=0;
    }

    return 0;
}