7-6 单源最短路径 (10 分)(最好用priirity_queue否则测试点2直接TLE)

请编写程序求给定正权有向图的单源最短路径长度。图中包含n个顶点,编号为0至n-1,以顶点0作为源点。

输入格式:

输入第一行为两个正整数n和e,分别表示图的顶点数和边数,其中n不超过20000,e不超过1000。接下来e行表示每条边的信息,每行为3个非负整数a、b、c,其中a和b表示该边的端点编号,c表示权值。各边并非按端点编号顺序排列。

输出格式:

输出为一行整数,为按顶点编号顺序排列的源点0到各顶点的最短路径长度(不含源点到源点),每个整数后一个空格。如源点到某顶点无最短路径,则不输出该条路径长度。

输入样例:

4 4
0 1 1
0 3 1
1 3 1
2 0 1

输出样例:

1 1 

 用的vector的邻接表存储,时间复杂度O(n^2),测试点2直接TLE了,建议换成priority_queue.

#include<bits/stdc++.h>
using namespace std;
#define MaxSize 20001
int dist[MaxSize];
bool visit[MaxSize]={false};
typedef struct enode
{
    int v,dis;
}Node;
vector<Node> v[MaxSize];
int vnum,e;
int main()
{
    cin>>vnum>>e;
    int i;
    for(i=0;i<e;i++){
        int p,q,length;
        cin>>p>>q>>length;
        Node temp;
        temp.v=q;
        temp.dis=length;
        v[p].push_back(temp);
    }
    fill(dist,dist+MaxSize,99999);
    dist[0]=0;
    int j;
    for(i=0;i<vnum;i++){
        int u=-1;
        int min=100000;
        for(j=0;j<vnum;j++){
            if(visit[j]==false && dist[j]<min){
                u=j;
                min=dist[j];
            }
        }
        if(u==-1){
            break;
        }
        visit[u]=true;
        for(j=0;j<v[u].size();j++){
            int q=v[u][j].v;
            if(visit[q]==false&& dist[u]+v[u][j].dis<dist[q]){
                dist[q]=dist[u]+v[u][j].dis;
            }
        }
    }
    for(i=1;i<vnum;i++){
        if(dist[i]!=99999){
            cout<<dist[i]<<" ";
        }
    }
    return 0;
}

 

 

上一篇:数据结构Java-顺序表


下一篇:发布自己写的Pop3协议组件