XTU online judge 1295 Flawless

Description
如果一个素数，依次去掉最高位得到一个数，这个数无前导0，并仍是素数的话，我们称其为“无瑕素数”。
比如317是素数，去掉最高位3得到17仍然是素数，再去掉最高位1得到7，仍然是素数，所以317是“无瑕素数”。
比如虽然107是素数，去掉最高位1得到7也是素数，但是因为存在前导0，所以这不是无瑕素数。
请写一个程序，判断某个素数是不是无瑕的。

Input
第一行是一个整数K ，表示样例的个数。 以后每行一个整数n(2≤n≤1,000,000,000) 。

Output
如果是无瑕素数，输出“Yes”，否则输出“No”。

Sample Input
3
3
107
317

Sample Output
Yes
No
Yes

解题思路：
首先判断数位是否有0，存在0肯定不是，然后依次去掉最高位判断是否为素数。注意别写太多循环，容易被超时卡成傻子。

#include<bits/stdc++.h>
using namespace std;

bool prime(int x)
{
	if(x == 1) return false;
	for(int i=2; i*i<=x; i++)
	{
		if(x%i == 0)
			return false;
	}
	return true;
}

bool first(int x)
{
	while(x)
	{
		if(x%10 == 0)
			return false;
		else
			x=x/10;
	}
	return true;
}

int length(int x)
{
	int len=0;
	while(x)
	{
		len++;
		x=x/10;
	}
	return len;
}

bool pd(int x)
{
	if(!prime(x)) return false;
	int cnt=length(x);
	for(int i=1,k=10; i<cnt; i++,k*=10)
	{
		if(!prime(x%k)) 
			return false;
	}
	return true;
}

int main()
{
	int k;
	cin>>k;
	
	while(k--)
	{
		int n;
		cin>>n;

		if(!pd(n) || !first(n))
			cout<<"No"<<endl;
		else 
			cout<<"Yes"<<endl;
	}
	return 0;
}