春天阳光灿烂,小宋突然心血来潮想骑自行车去学校外面玩,但是到达目的地的路线不止一条,他想尽快的到达目的地,又能不需要骑太远的路, 你能帮助他吗?
输入格式:
输入包含一个测试数据,第一行有三个整数n(2 <= n <= 1000),途中可能经过的地点的个数,地点编号1~n;m(1 <= m <= 10000),为路径的条数;d(2 <= d <= n),目的地编号;其中学校为起点,默认为1。接下来m行: x y time dist , x y表示地点x,y是可以相互到达的,time,dist分别表示x到y或y到x的时间,距离。
输出格式:
按如下格式输出“花费的时间+空格+要骑的距离+空格+从学校到达目的地的路径”,路径中的两个地点之间以箭头(->)分隔。(具体见输出样例)
输入样例:
在这里给出一组输入。例如:
7 8 7
1 2 1 1
1 3 1 1
2 4 1 2
3 4 1 1
4 5 1 2
4 6 1 1
5 7 1 1
6 7 2 1
输出样例:
在这里给出相应的输出。例如:
4 5 1->3->4->5->7
代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB
栈限制
8192 KB
堆优化版dijkstra,存储路径+多权值问题
#include <bits/stdc++.h>
using namespace std;
typedef pair<pair<int,int>,int> PII;
const int N = 1e5+5;
int h[N], e[N], w1[N], w2[N], ne[N], idx;
int n, m, rt, d[N], t[N], p[N];
int st[N];
void add(int a, int b, int c, int d)
{
e[idx] = b, w1[idx] = c, w2[idx] = d;
ne[idx] = h[a], h[a] = idx++;
}
void print(int x)
{
if(p[x] != -1) print(p[x]);
if(p[x] == -1) cout<<x;
else cout<<"->"<<x;
}
void dij()
{
memset(d, 0x3f, sizeof d);
memset(t, 0x3f, sizeof t);
priority_queue<PII, vector<PII>, greater<PII>> q;
q.push({{0, 0}, 1});
d[1] = 0, t[1] = 0, p[1] = -1;
while(q.size())
{
auto tem = q.top();q.pop();
int ver = tem.second, tm = tem.first.first, dt = tem.first.second;
if(st[ver]) continue;
st[ver] = 1;
for(int i=h[ver]; ~i; i=ne[i])
{
int j = e[i];
if(t[j] > tm + w1[i])
{
t[j] = tm + w1[i];
d[j] = dt + w2[i];
p[j] = ver;
q.push({{t[j], d[j]}, j});
}
else if(t[j] == tm + w1[i])
{
if(d[j] > dt + w2[i])
{
d[j] = dt + w2[i];
p[j] = ver;
q.push({{t[j], d[j]}, j});
}
}
}
}
cout<<t[rt]<<' '<<d[rt]<<' ';
print(rt);
}
int main()
{
memset(h, -1, sizeof h);
cin>>n>>m>>rt;
while(m--)
{
int x, y, tm, dt;
cin>>x>>y>>tm>>dt;
add(x, y, tm, dt);
add(y, x, tm, dt);
}
dij();
puts("");
return 0;
}