题意: n个点(0~n-1),m条边,每条边有两个权值,一个是时间,一个是花费,其中0是首都,要让0点到其他所有点的总时间最短,同时要让花费最小。
二重限制的最短路,稍微改一下即可。
关于正确性的一些思考:
对prim算法有一点理解的话,容易发现dij和prim本质和过程都无比类似,每次选一个点拓展,区别在于dij维护其他点到源点的最小距离,prim维护其他点到MST集合的最小距离。
因此此题只需要同时用到dijkstra和prim类似的过程,在维护最短距离的前提下维护最小的花费即可解决,可以理解成每次拓展一个点的时候,在dij中会找到多条边,prim就用在些边中选择最小花费。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int M = 1e5+5;
const ll llinf = ~0ull>>2;
struct eg{
int u, v;
ll cost1, cost2;
eg(){}
eg(int a, int b, ll c, ll d){ u = a, v = b, cost1 = c, cost2 = d; }
};
vector<eg>G[M];
void add(int a, int b, ll c, ll d){
G[a].push_back(eg(a, b, c, d));
G[b].push_back(eg(b, a, c, d));
}
struct node{
int u;
ll d, c;
node(int _u, ll _d, ll _c){ u = _u, d = _d, c = _c; }
bool operator < (const node& a) const{
if(d == a.d) return c > a.c;
return d > a.d;
}
};
ll dis[M], cost[M]; //dis是dij过程,cost是prim过程
bool vis[M];
void exdijkstra(int s, int n){
for(int i = 0; i <= n; ++i) dis[i] = cost[i] = llinf, vis[i] = 0;
dis[s] = 0, cost[s] = 0;
priority_queue<node>q;
q.push(node(s, 0, 0));
ll ans1 = 0, ans2 = 0;
for(int tt = 0; tt < n; ++tt){
while(!q.empty() && q.top().d != dis[q.top().u] || q.top().c != cost[q.top().u]) q.pop();
int u = q.top().u; ll d = q.top().d; ll c = q.top().c; q.pop();
vis[u] = 1;
ans1 += d, ans2 += c;
for(int k = 0; k < G[u].size(); ++k){
int v = G[u][k].v, nd = G[u][k].cost1, nv = G[u][k].cost2;
if(!vis[v] && dis[v] > d + nd || (dis[v] == d+nd && cost[v] > nv)){ //优先维护距离
dis[v] = d + nd;
cost[v] = nv;
q.push(node(v, dis[v], cost[v]));
}
}
}
printf("%lld %lld\n", ans1, ans2);
}
int main(){
int T;
scanf("%d", &T);
while(T--){
int n, m;
scanf("%d%d", &n, &m);
for(int i = 0; i <= n; ++i) G[i].clear();
for(int i = 0; i < m; ++i){
int a, b;
ll c, d;
scanf("%d%d%lld%lld", &a, &b, &c, &d);
add(a, b, c, d);
}
exdijkstra(0, n);
}
}