poj 1511-- Invitation Cards （dijkstra+优先队列）

最新推荐文章于 2019-08-24 11:30:00 发布

只会签到题

最新推荐文章于 2019-08-24 11:30:00 发布

阅读量101

点赞数

本文链接：https://blog.csdn.net/weixin_43735124/article/details/106271955

版权

本文介绍了一种使用两次Dijkstra算法结合优先队列解决有向图中志愿者到站点往返开销最小化问题的方法。通过将早晨和晚上的路径分别建模并求解，实现了开销的有效计算。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

刚开始想复杂了，一直做不出来，，，其实就是两遍dijkstra+优先队列（其实就是板子题，只要能有个好的板子，剩下的都不是事），做出来感觉好简单......

题意：有n个车站和n个志愿者，早上每个志愿者去一个站点，晚上回去，问最少的开销是多少。是一个有向图

先一遍dijkstra求出早上的开销，在把车站倒过来及1到4变成4到1，然后再一遍dijkstra求出晚上的开销。

不要用memset去初始化为INF，本题如果用memset会wa，记得开long long；

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<queue>
 4 using namespace std;
 5 
 6 const int INF = 1e9+10;
 7 const int N = 1000010;
 8 const int M = 1000010;
 9 int tot;
10 
11 struct Edge                      //dijkstra＋优先队列板子
12 {
13     int v, cost, next;
14 }edge[M];
15 
16 struct qnode
17 {
18     int v,c;
19     bool operator < (const qnode &x) const {
20         return c > x. c;
21     }
22 };
23 
24 int head[M];
25 bool vis[N];
26 int dis[N];
27 
28 void addedge(int u, int v, int w)
29 {
30     edge[tot]. v = v;
31     edge[tot]. cost = w;
32     edge[tot]. next = head[u];
33     head[u] = tot ++;
34 }
35 
36 void dijkstra(int st)
37 {
38     memset(vis, 0, sizeof(vis));
39     for(int i=0;i<N;i++)
40         dis[i]=INF;                 //最好不要用memset去初始化为INF， 很容易wa
41     dis[st] = 0;
42     priority_queue <qnode> q;
43     q. push({st, 0});
44     while(! q. empty()){
45         qnode t = q. top();
46         q. pop();
47         if(vis[t. v])
48             continue;
49         vis[t. v] = true;
50         for(int i = head[t. v]; i != -1; i = edge[i]. next){
51             int v = edge[i]. v;
52             int cost = edge[i]. cost;
53             if(! vis[v] && dis[v] > dis[t. v] + cost){
54                 dis[v] = dis[t. v] + cost;
55                 q. push({v, dis[v]});
56 
57             }
58         }
59     }
60 }
61 
62 int a[N], b[N], c[N];
63 
64 int main(){
65     int n, m,t;
66     scanf("%d",&t);
67     while(t--){
68         scanf("%d%d", &n, &m);
69         tot = 1;
70         memset(edge,0,sizeof edge);
71         memset(head, -1, sizeof(head));
72         for(int i=0;i<m;i++){
73             scanf("%d%d%d", &a[i], &b[i], &c[i]);
74             addedge(a[i], b[i], c[i]);
75         }
76         dijkstra(1);                        //早上的
77         int sum=0;
78         for(int i=2;i<=n;i++){
79             sum+=dis[i];                    //把1到每个站的最小开销加起来
80         }
81         tot=1;
82         memset(edge,0,sizeof edge);
83         memset(head, -1, sizeof(head));
84         for(int i=0;i<m;i++){
85             addedge(b[i], a[i], c[i]);      //正着是1到每个站，反过来就是每个站到1
86         }
87         dijkstra(1);                        //晚上的
88         for(int i=2;i<=n;i++){
89             sum+=dis[i];                    //把每个站到1的最小开销加起来
90         }
91         printf("%d\n",sum);
92     }
93     return 0;
94 }