团体程序设计天梯赛-练习集

Posted 2020-12-04 tingtin

 1 L2-001 紧急救援（25 分）
 2 
 3 #include <iostream>
 4 #include <cstdio>
 5 #include <algorithm>
 6 #include <cstring>
 7 #include <vector>
 8 #include <utility>
 9 #include <queue>
10 using namespace std;
11 typedef pair<int,int>P;
12 int n,m,s,d;
13 const int  N=550;
14 const int inf=0x3f3f3f3f;
15 int val[N],dis[N],pre[N],prenum[N];
16 int tval[N];
17 struct Node{
18     int  to,w;
19     Node(){}
20     Node(int TO,int W){
21         to=TO;
22         w=W;
23     }
24 };
25 vector<Node>vec[N];
26 void bfs(int  sta){
27     priority_queue<P,vector<P>,greater<P> >que;
28     que.push(P(0,sta));//此时的优先队列会按照first的值排序
29     prenum[sta]=1;//1,不是0
30     tval[sta]=val[sta];
31     dis[sta]=0;
32     while(!que.empty()){
33         P q=que.top();
34         que.pop();//容易忘记，弹出队列
35         int v=q.second;
36         if(dis[v]<q.first) continue;
37         for(int i=0;i<vec[v].size();i++){
38             Node Nod=vec[v][i];
39             int t=Nod.to;
40             if(dis[t]>dis[v]+Nod.w){
41                 dis[t]=dis[v]+Nod.w;
42                 pre[t]=v;//记录最短路上的该点的前一个点
43                 tval[t]=tval[v]+val[t];//救援队数目是依次累加的
44                 prenum[t]=prenum[v];
45                 que.push(P(dis[t],t));//容易忘记，压入队列
46             }
47             else if(dis[t]==dis[v]+Nod.w){
48                 prenum[t]+=prenum[v];//到t的路径数目要加上到v的路径数目
49                 if(tval[t]<tval[v]+val[t]){//只有救援队数目多了，才会更新节点
50                     tval[t]=tval[v]+val[t];
51                     pre[t]=v;    
52                 }
53             }
54         }
55     }    
56 }
57 int  main()
58 {
59     scanf("%d%d%d%d",&n,&m,&s,&d);
60     for(int i=0;i<n;i++)
61         scanf("%d",&val[i]);
62     int  x,y,z;
63     for(int  i=0;i<m;i++){
64         scanf("%d%d%d",&x,&y,&z);
65         vec[x].push_back(Node(y,z));
66         vec[y].push_back(Node(x,z));//本题为无向图
67     }
68     memset(pre,-1,sizeof(pre));//不能初始化为0，0也是节点
69     memset(dis,inf,sizeof(dis));
70     bfs(s);
71     printf("%d %d
",prenum[d],tval[d]);
72     int ans=0;
73     int ret[N];
74     int cur=d;
75     while(cur!=-1){
76         ret[ans++]=cur;
77         cur=pre[cur];
78     }
79     for(int i=ans-1;i>=0;i--){
80         printf("%d%c",ret[i],i==0?‘
‘:‘ ‘);
81     }
82     return   0;
83 }