poj 3026Borg Maze

http://poj.org/problem?id=3026

 #include<cstdio>
 #include<iostream>
 #include<cstring>
 #include<algorithm>
 #define maxn 100
 #define maxn1 100000
 #include<queue>
 using namespace std;

 char s[maxn][maxn];
 const int inf=<<;
 int n,m;
 bool vis[maxn*][maxn*];
 bool visi[maxn*];
 int dis[maxn*][maxn*];
 int c[maxn*][maxn*];
 int dir[][]={{,},{-,},{,},{,-}};
 int dist[maxn*];
 int ans;

 struct node
 {
     int x,y;
     int step;
 }p[maxn1],st1,st;

 void bfs(struct node st2)
 {
     st2.step=;
     queue<node>q;
     q.push(st2);
     memset(vis,false,sizeof(vis));
     vis[st2.x][st2.y]=true;
     while(!q.empty())
     {
         st1=q.front();
         q.pop();
         for(int i=; i<; i++)
         {
             int xx=st1.x+dir[i][];
             int yy=st1.y+dir[i][];
             if(xx>=&&xx<m&&yy>=&&yy<n&&!vis[xx][yy]&&s[xx][yy]!='#')
             {
                 if(s[xx][yy]=='S'||s[xx][yy]=='A')
                 {
                     dis[c[st2.x][st2.y]][c[xx][yy]]=st1.step+;
                     st.x=xx;
                     st.y=yy;
                     st.step=st1.step+;
                     q.push(st);
                     vis[xx][yy]=true;
                 }
                 else
                 {
                     st.x=xx;
                     st.y=yy;
                     st.step=st1.step+;
                     q.push(st);
                     vis[xx][yy]=true;
                 }
             }
         }
     }
 }

 bool prime(int num)
 {
      memset(visi,false,sizeof(visi));
      for(int i=; i<=num; i++)
      {
          dist[i]=inf;
      }
      ans=;dist[]=;
      for(int i=; i<num; i++)
      {
          int temp=inf,k=;
          for(int j=; j<num; j++)
          {
              if(!visi[j]&&dist[j]<temp)
              {
                  temp=dist[j];
                  k=j;
              }
          }
          if(temp==inf) return false;
          visi[k]=true;
          ans+=temp;
          for(int j=; j<num; j++)
          {
              if(!visi[j]&&dist[j]>dis[k][j])
                 dist[j]=dis[k][j];
          }
      }
      return true;
 }

 int main()
 {
     int t;
     scanf("%d",&t);
     while(t--)
     {
         int num=;
         char s1[];
         scanf("%d%d",&n,&m);
         gets(s1);
         memset(vis,false,sizeof(vis));
         memset(c,-,sizeof(c));
         for(int i=; i<m; i++)
         {
             gets(s[i]);
             for(int j=; j<n; j++)
             {
                 if(s[i][j]=='A'||s[i][j]=='S')
                 {
                     p[num].x=i;
                     p[num].y=j;
                     c[i][j]=num;
                     num++;
                 }
             }
         }
         for(int i=; i<num; i++)
         {
             bfs(p[i]);
         }
         prime(num);
         printf("%d\n",ans);
     }
     return ;
 }