Happy Happy Prime Prime UVALive - 7094

Happy Happy Prime Prime

题意：
给出一个n，判断n是不是素数，不是，输出no，是，继续判断经过循环，平方迭代最后能否等于1，可以，输出yes，不可以，输出no;
思路：
判断是不是素数，判断之后用两个while进行循环迭代，由于做的时候没考虑，如果最后n不等于1的时候该怎么办，从而导致超时，且如果n不等于1，n会迭代到出现过的n,所以利用一个数组标记n；
代码：

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cmath>
using namespace std;
int vis[10005];
int pan(int n)
{
    int faul=1;
    for(int i=2; i<=sqrt(n); i++)
    {
        if(n%i==0)
        {
            faul=0;
            break;
        }
    }
    if(faul==1)
    {
        return 1;
    }
    else
    {
        return 0;
    }
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int w;
        int m;
        int flag=0;
        scanf("%d%d",&w,&m);
        if(pan(m)==0||m==1)
        {
            printf("%d %d NO\n",w,m);
        }
        else
        {

            memset(vis,0,sizeof(vis));
            int n=m;
            int p;
            while(!vis[n]&&n!=1)
            {
                p=0;
                vis[n]=1;
                while(n>=1)
                {
                    p=p+(n%10)*(n%10);
                    n=n/10;
                }
                if(p!=1)
                {
                    n=p;
                }
                else
                {
                    flag=1;
                    break;
                }

            }
            if(flag==1)
            {
                printf("%d %d YES\n",w,m);
            }
            else
            {
                printf("%d %d NO\n",w,m);
            }



        }
    }
}

总会一点一点进步。