CCF交通规划

很郁闷devc++编译器上跑起来是对的，但交上去就是运行错误

问题描述

　　G国国王来中国参观后，被中国的高速铁路深深的震撼，决定为自己的国家也建设一个高速铁路系统。
　　建设高速铁路投入非常大，为了节约建设成本，G国国王决定不新建铁路，而是将已有的铁路改造成高速铁路。现在，请你为G国国王提供一个方案，将现有的一部分铁路改造成高速铁路，使得任何两个城市间都可以通过高速铁路到达，而且从所有城市乘坐高速铁路到首都的最短路程和原来一样长。请你告诉G国国王在这些条件下最少要改造多长的铁路。

输入格式

　　输入的第一行包含两个整数n, m，分别表示G国城市的数量和城市间铁路的数量。所有的城市由1到n编号，首都为1号。
　　接下来m行，每行三个整数a, b, c，表示城市a和城市b之间有一条长度为c的双向铁路。这条铁路不会经过a和b以外的城市。

输出格式

　　输出一行，表示在满足条件的情况下最少要改造的铁路长度。

样例输入

4 5
1 2 4
1 3 5
2 3 2
2 4 3
3 4 2

样例输出

11

评测用例规模与约定

　　对于20%的评测用例，1 ≤ n ≤ 10，1 ≤ m ≤ 50；
　　对于50%的评测用例，1 ≤ n ≤ 100，1 ≤ m ≤ 5000；
　　对于80%的评测用例，1 ≤ n ≤ 1000，1 ≤ m ≤ 50000；
　　对于100%的评测用例，1 ≤ n ≤ 10000，1 ≤ m ≤ 100000，1 ≤ a, b ≤ n，1 ≤ c ≤ 1000。输入保证每个城市都可以通过铁路达到首都。

思路：1、要求改造后其他顶点m到顶点1的最短距离与改造前的一样长，最小树，各顶点连通最少需要n-1条边

            2、先求单源最短路径生成拓扑图，再求每个节点x到其各个先驱节点的最小距离的和为最后答案

#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <cstdio>
#include <string>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <string.h>
using namespace std;
//单源最短路+并查集
const int MAXN=10000+10;
const int MAXM=100000+10;
const int INF = 0x3f3f3f3f;

int n,m,ma[MAXN][MAXN],vist[MAXN];
vector<int> arr[MAXN];//arr[i]中第一个位置即arr[i][0]存放顶点i到顶点1的距离，其他存放i的前驱结点
void read()
{
    cin>>n>>m;
    memset(ma,INF,sizeof(ma));
    memset(vist,0,sizeof(vist));//0代表该节点未被访问过 
    for(int i=2;i<MAXN;i++)
    arr[i].push_back(INF);
    int a,b,len;
    for(int i=0;i<m;i++)
    {
        cin>>a>>b>>len;
        ma[a][b]=len;
		ma[b][a]=len;
		if(a==1 || b==1)
		{
			int x;
			if(b==1)
			x=a;
			else
			x=b;
			arr[x].clear();
			arr[x].push_back(len);
			arr[x].push_back(1);
		} 
    }
    return;
}
void dij()
{
	vist[1]=1;
	do{
		int max=INF,now=0;
		for(int i=2;i<=n;i++)
		{
			if(vist[i]==0 && arr[i][0]<max)
			{
				max=arr[i][0];
				now=i;
			}
		}
		if(now==0)
		break;
		vist[now]=1;//访问过
		for(int i=2;i<=n;i++)
		{
			if(vist[i]==1)
			continue;
			if(arr[i][0]>arr[now][0]+ma[now][i])
			{
				arr[i].clear();
				arr[i].push_back(arr[now][0]+ma[now][i]);
				arr[i].push_back(now);
			}
			else if(arr[i][0]==arr[now][0]+ma[now][i])
			{
				arr[i].push_back(now);
			}
			else
			{
				continue;
			}
		}	 
	}while(true);
	return;
}
int unio()
{
	int ans=0;
	for(int i=2;i<=n;i++)
	{
		int x=INF;
		for(int j=1;j<arr[i].size();j++)
		{
			if(ma[i][arr[i][j]]<x)
			{
				x=ma[i][arr[i][j]];
			}
		}
		ans+=x;
	}
	return ans;
	
} 
int main()
{
    read();
    dij();
    cout<<unio()<<endl; 
    return 0;
}