杭电3371 prime（）算法

蛋疼的G++，，不解释。不过最近看楼教主似乎提交的代码都是用G++的，，，，看来以后提交代码都用G++吧，，，，能提高水平。。。。。用prime算法过的，，，跑了900多ms，，，无语，，，求优化！

G++ac代码：

#include <iostream>
#include <string.h>
#include <algorithm>
using namespace std;
const int N=502;
//const int INF=0xfffffff;
#define INF 10005
int visted[N],map[N][N],key[N];
int n,sum;
void prime(){
  memset(visted,0,sizeof(visted));
  for(int i=1;i<=n;++i)
      key[i]=map[1][i];
  key[1]=0;visted[1]=1;
  int temp,k;
  sum=0;
  for(int i=1;i<=n;++i){
      temp=INF;
      for(int j=1;j<=n;++j){
          if(!visted[j]&&temp>key[j]){
            temp=key[j];k=j;
          }
      }
      if(temp==INF)
          break;
      //printf("temp===%d\n",temp);
      visted[k]=1;
      sum+=temp;
      //printf("sum==%d\n",sum);
      for(int j=1;j<=n;++j){
        if(!visted[j]&&key[j]>map[k][j])
            key[j]=map[k][j];
      }
  }
}
int main(){
  int kk;
  scanf("%d",&kk);
  while(kk--){
    int m,k;
    memset(map,INF,sizeof(map));
    scanf("%d%d%d",&n,&m,&k);
    int a,b,c;
    while(m--){
      scanf("%d%d%d",&a,&b,&c);
      if(c<map[a][b])
          map[a][b]=map[b][a]=c;
    }
    int t,x,y;
    while(k--){
      scanf("%d",&t);
      scanf("%d",&x);
      t--;
      while(t--){
        scanf("%d",&y);
        map[x][y]=map[y][x]=0;
      }
    }
    prime();
    int flag=1;
    for(int i=1;i<=n;++i){
        if(visted[i]==0){
          flag=0;break;
        }
    }
    if(flag)
        printf("%d\n",sum);
    else
        printf("-1\n");
  }
  return 0;
}