Prime Path(my last ac)

http://acm.hdu.edu.cn/diy/contest_showproblem.php?pid=1004&cid=906

该题关键是把图构出来，但得到邻接矩阵后，想用bfs来搜索，发现超时很严重，于是只能试着用dij来做，居然AC了，嘿嘿~begin from search,end to search……

 

  1#include <iostream>
  2#include <queue>
  3#include <math.h>
  4using namespace std;
  5#define INF 1000000000
  6int arr[1100][1100];
  7int p[1100];
  8bool fun(int n,int m)//判断两个数是否只有一位相等
  9{
 10    int same=0,t=4;
 11    while(t--)
 12    {
 13        if(n%10==m%10) same++;
 14        n/=10;
 15        m/=10;
 16    }
 17    if(same==3) return 1;
 18return 0;
 19}
 20int prime(int n)
 21{
 22    int max=(int)sqrt((double)n)+1;
 23    int i;
 24    for (i=3;i<=max;i+=2)
 25    {
 26        if(n%i==0)return false;
 27    }
 28    return true;
 29}
 30
 31void initprime()
 32{
 33    int i,t=1;
 34    for(i=1001;i<=9999;i+=2)
 35    {
 36        if (prime(i))
 37        {
 38            p[t++]=i;
 39        }
 40    }
 41}
 42void getarr()//构建邻接矩阵
 43{
 44    int i,j;
 45    for (i=1;i<=1061;i++)
 46    {
 47        for (j=i+1;j<=1061;j++)
 48        {
 49            if (fun(p[i],p[j]))
 50            {
 51                arr[i][j]=arr[j][i]=true;
 52            }
 53        }
 54    }
 55}
 56int getindex(int n)//查找n的索引值
 57{
 58    int i=1,j=1061,mid;
 59    while(1)
 60    {
 61        if(mid!=(i+j)/2)
 62            mid=(i+j)/2;
 63        else
 64        {
 65            if(p[i]==n)
 66                return i;
 67            if(p[j]==n)
 68                return j;
 69        }
 70        if(p[mid]==n)
 71            return mid;
 72        if (p[mid]>n)
 73        {
 74            j=mid;
 75        } 
 76        else
 77        {
 78            i=mid;
 79        }
 80    }
 81}
 82// void bfs(int m,int n)
 83// {
 84//     queue<int>coll;
 85//     queue<int>q;//存储连接的个数
 86//     int i,index,s=1;
 87//     bool doit=false;
 88//     coll.push(m);
 89//     q.push(s);
 90//     while(!coll.empty())
 91//     {
 92//         index=coll.front();
 93//         s=q.front();
 94//         coll.pop();
 95//         q.pop();
 96//         for (i=0;i<1061;i++)
 97//         {
 98//             if(arr[index][i]==1)
 99//             {
//                 if (i==n)//找到
//                 {
//                     doit=true;
//                     break;
//                 }
//                 else
//                 {
//                     coll.push(i);
//                     q.push(s+1);
//                 }
//             }
//         }
//         if (doit)
//         {
//             break;
//         }
//     }
//     if(doit)
//         printf("%d\n",s);
//     else
//         printf("Impossible\n");
// }
int dijkstra(int s, int t, int n)
{    
    int i, m[1100];    
    int v, min_value;    
    int visit[1100];    
    for(i = 1; i <= n; i ++)        
    {        
        m[i] = INF;    
        visit[i] = 0;        
    }    
    m[s] = 0;    
    while( 1 )        
    {        
        v = -1; min_value = INF;         
        for(i = 1; i <= n; i ++)            
        {            
            if(visit[i] == 0){                
                if(m[i] < min_value)
                {                    
                    min_value = m[i];                    
                    v = i;                    
                }                
            }            
        }        
        if(v == -1 || v == t)break;        
        visit[v] = 1;        
        for(i = 1; i <= n; i ++)            
        {            
            if(visit[i] == 0)                
            {                
                if(m[v] + arr[v][i] < m[i])                    
                {                    
                    m[i] = m[v] + arr[v][i];
                    
                }                
            }            
        }        
    }    
    return m[t];    
}
int main()
{
    int n,a,b,x;
    memset(arr,1,sizeof(arr));
    initprime();
    getarr();
    scanf("%d",&n);
    while(n--)
    {
        scanf("%d %d",&a,&b);
        if(a==b)
            printf("0\n");
        else
        { 
            int i=getindex(a);
            int j=getindex(b);
            x=dijkstra(i,j,1061);
            if(x!=INF)
                printf("%d\n",x);
            else
                printf("Impossible\n");
        }
    }
    return 0;
}