Aizu - 2249 Road Construction 最短路变形

传送门：AOJ 2249

题意：出若干个建筑之间的一些路，每条路都有对应的长度和需要的花费，问在保证源点1 到其他个点的距离最短的情况下，最少的花费是多少。

思路：以前做过类似的题，就是在松弛过程中维护花费，如果相等的话特殊判断一下就行，然而我把维护cost的过程写成了和维护dis一样的松弛操作。。应该是直接赋值就好了，不用加上前面的cost，否则就会有很多边被重复计算。

代码：

#include<bits/stdc++.h>
#define ll long long
#define pb push_back
#define fi first
#define se second
#define pi acos(-1)
#define inf 0x3f3f3f3f
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define rep(i,x,n) for(int i=x;i<n;i++)
#define per(i,n,x) for(int i=n;i>=x;i--)
using namespace std;
typedef pair<int,int>P;
const int MAXN=100010;
int gcd(int a,int b){return b?gcd(b,a%b):a;}
struct node{
	int v, d, c;
	node(int _v, int _d, int _c) : v(_v), d(_d), c(_c){
	}
};
vector<node> mp[MAXN];
int dis[MAXN], cost[MAXN];
bool book[MAXN];
queue<int>q;
void spfa(int n)
{
	memset(dis, inf, sizeof(dis));
	memset(book, 0, sizeof(book));
	memset(cost, inf, sizeof(cost));
	book[1] = 1;
	dis[1] = 0;
	cost[1] = 0;
	q.push(1);
	while(!q.empty())
	{
		int u = q.front();
		q.pop();
		for(int i = 0; i < mp[u].size(); i++)
		{
			node &e = mp[u][i];
			if(dis[e.v] > dis[u] + e.d)
			{
				dis[e.v] = dis[u] + e.d;
				cost[e.v] = e.c;
				if(!book[e.v])
				{
					book[e.v] = 1;
					q.push(e.v);
				}
			}
			else if(dis[e.v] == dis[u] + e.d && cost[e.v] > e.c)
			{
				cost[e.v] = e.c;
				if(!book[e.v])
				{
					book[e.v] = 1;
					q.push(e.v);
				}
			}
		}
		book[u] = 0;
	}
	int ans = 0;
	for(int i = 1; i <= n; i++)
	ans += cost[i];
	cout << ans << endl;
}
int main()
{
	int n, m, u, v, d, c;
	while(cin >> n >> m, n + m)
	{
		for(int i = 0; i < m; i++)
		{
			scanf("%d %d %d %d", &u, &v, &d, &c);
			mp[u].pb(node(v, d, c));
			mp[v].pb(node(u, d, c));
		}
		spfa(n);
		for(int i = 1; i <= n; i++)
		mp[i].clear();
	}
 	return 0;
}