hdu 3371 Connect the Cities

链接：点击打开链接

连接城市，给你M个城市，有M条道路，还有K个城市相连，接下来M行，P与Q城市相连花费C，然后接下来K行,首先有一个数T，代表有T个城市相连，然后给胡相连城市的ID。

简单的prim，不过用C++提交可以过，但用G++,就会超时，不超时要自己写一个scanf函数。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int INF=0xffffff;
int edge[510][510];
int nn,vis[510],m,k;
void prime(int x){
 int i,j,k;
 int sum_i,mark[510],mi_n;
 for(i=1;i<=nn;i++){
  mark[i]=edge[x][i];
  vis[i]=0;
 }
 vis[x]=1;
 sum_i=0;
 for(i=2;i<=nn;i++){
   mi_n=INF;
   for(j=1;j<=nn;j++){
    if(!vis[j]&&mark[j]<mi_n){
    mi_n=mark[j];
    k=j;
    }
   }
   vis[k]=1;
   sum_i+=mi_n;
   if(sum_i>=INF)
   break;
   for(j=1;j<=nn;j++){
    if(!vis[j]&&edge[k][j]<mark[j])
    mark[j]=edge[k][j];
   }
 }
 if(sum_i>=INF)
 printf("-1\n");
 else
 printf("%d\n",sum_i);
}
inline int scanf(int &num)
{
    char in;
    bool neg=false;
    while((in=getchar())!=EOF && (in>'9' || in<'0') && in!='-');
    if(in==EOF) return 0;
    else if(in=='-') neg=true,num=0;
    else  num=in-'0';
    while(in=getchar(),in>='0' && in<='9') num*=10,num+=in-'0';
    if(neg) num=-num;
    return 1;
}
int main(){
 int i,j,t,a,b,c,d,x[510];
 scanf("%d",&t);
 while(t--){
    scanf(nn);
    scanf(m);
    scanf(k);
    for(i=1;i<=nn;i++)
    for(j=1;j<=nn;j++)
    edge[i][j]=INF;
  for(i=1;i<=m;i++){
  scanf(a);
  scanf(b);
  scanf(c);
  if(edge[a][b]>c){
  edge[a][b]=c;
  edge[b][a]=c;
  }
  }
  for(i=1;i<=k;i++){
   scanf(d);
   for(j=1;j<=d;j++)
   scanf(x[j]);
   for(j=1;j<d;j++)
     edge[x[j]][x[j+1]]=edge[x[j+1]][x[j]]=0;
    }
   prime(1);
 }
 return 0;
}