poj 3126 Prime path

題目銜接：http://poj.org/problem?id=3126

題目大意：對一個4位數進行變換，每次只能變一個數字，求變化到另一數字的最小變換數，注意第一位不能爲0，用寬度優先搜索，第一次找到該數字時，即爲最小變換次數

代碼如下：

#include <iostream>
#include <queue>
#include <cstring>

using namespace std;
#define maxn 10000

typedef struct data{
	int num;
	int pound;
}data;

int first,second;
bool visited[maxn];

bool is_prime(int num)
{
	for(int i=2;i<num/2;i++)
		if(num%i == 0)
			return false;
	return true;
}

int get_next_prime(int num)
{
	for(int i=0;i<=9;i++){
		
		int m = (num/10)*10 + i;   //改變個位
		if(!visited[m] && is_prime(m))  //注意2個條件的順序，可以減少計算量
			return m;
		
		
		int n = (num/100)*100+num%10+i*10;   //改變十位
		if(!visited[n] && is_prime(n))
			return n;
		
		
		int p = (num/1000)*1000+i*100+((num%100)/10)*10+num%10;  //改變百位
		if(!visited[p] && is_prime(p))
			return p;
		if(i != 0){  //首位不能爲0
		int q = i*1000+((num%1000)/100)*100+((num%100)/10)*10+num%10;  //改變千位
		
		if(!visited[q] && is_prime(q))
			return q;
		}
	}

	return -1;  //如果沒有結果，返回-1

}

int answer_bfs()
{
	queue<data> store;
	data first_item;
	first_item.num = first;
	first_item.pound = 0;
	store.push(first_item);
	visited[first]=true;

	if(first == second)
		return 0;

	while(!store.empty()){
		data temp_data = store.front();
		store.pop();
		int number = temp_data.num ;
		int prime = get_next_prime(number);
		while(prime != -1){
		visited[prime] = true;
		data temp;
		temp.num = prime;
		 
		temp.pound = temp_data.pound+1;
		if(prime == second)                 //找到結果，直接返回
			return temp.pound;
		store.push(temp);
		prime = get_next_prime(number);
		}
	}
	return -1;
}

int main()
{
	int cases;
	cin>>cases;
	while(cases--){
		memset(visited,false,sizeof(visited));  //記住初始化
		cin>>first>>second;
		int answer = answer_bfs();
		if(answer == -1)
			printf("Impossible\n");
		else
			printf("%d\n",answer);
	}
	return 0;
}