POJ 3126 Prime Path

Prime Path

BFS。题目大意：给定两个个数(4位数)a ,b ，用0~9置换a中的一位，然后得到一个新数，要求这个新数是素数，然后对这个新数同样操作，直到得到的数与b相同，求最少的步骤。直接BFS就好，由于是4位数，所以先打表求出4位数中的所有的素数，要求，最高位不为0，这一点可以由第二组数据得到。

#include<stdio.h>
#include<string.h>

#define MAXN 10005
bool prim[MAXN] ;
int  vis[MAXN] ;
int  queue[MAXN] ;
int front ;
int rear  ;
int d[4]  ;

void pop(){
	front = (front + 1) % MAXN ;
}

void push(int l){
	queue[rear] = l ;
	rear = (rear + 1) % MAXN ;
}

int getFront(){
	return queue[front] ;
} 
bool empty(){

	if(rear == front)
		return 1 ;
	return 0 ;	 
}

void createprim(){

	int i ;
	int j ;

	for(i = 1 ; i < MAXN ; i ++)
		prim[i] = 1 ;
	prim[1] = 0 ;

	for(i = 2 ; i < MAXN ; i ++){
		if(prim[i]){
			for(j = 2 ; i * j < MAXN ; j ++){
				prim[i * j] = 0 ;
			} 
		}
	}
	return ; 
}

void next(int x){
	d[3] = x / 1000 ;
	d[2] = (x / 100) % 10 ;
	d[1] = (x % 100) / 10 ;
	d[0] = x % 10 ;  
}

void bfs(int s , int r){
	
	front = 0 ;
	rear = 0 ;
	
	vis[s] = 1 ;

	push(s) ;
	int a ;
	int b ;

	while(!empty()){
		a = getFront() ;
		pop() ;

		if(a == r){
			printf("%d\n" , vis[a] - 1) ;
			return ;			
		}

		next(a) ;
		for(int i = 0 ; i < 4 ; i ++){
			
			for(int j = 1 ; j <= 10 ; j ++){
				
				d[i] = (++ d[i]) %10 ;

				if(!d[3])
					continue ;
				
				b = d[3] * 1000 + d[2] * 100 + d[1] * 10 + d[0] ; 
			
				if(prim[b] && !vis[b]){
				
					if(b == r){
						printf("%d\n" , vis[a] ) ;
						return ;
					}
				
					vis[b] = vis[a] + 1 ;
					push(b) ;
				}	
			}
		}
	}

	printf("Impossible\n") ;
	return ;
}

int main(){
	int t ;
	scanf("%d" , &t) ;
	
	int a ;
	int b ;
	
	createprim() ;

	while(t--){

		memset(vis , 0 , sizeof(vis)) ;
 		scanf("%d %d" , &a , &b) ;
		bfs(a , b) ;
	}

	return  0 ;
}