hdoj3790+优先队列结构体重载优化的dij
链接:http://acm.hdu.edu.cn/showproblem.php?pid=3790
题目:
最短路径问题
Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 42431 Accepted Submission(s): 12184
Problem Description
给你n个点,m条无向边,每条边都有长度d和花费p,给你起点s终点t,要求输出起点到终点的最短距离及其花费,如果最短距离有多条路线,则输出花费最少的。
Input
输入n,m,点的编号是1~n,然后是m行,每行4个数 a,b,d,p,表示a和b之间有一条边,且其长度为d,花费为p。最后一行是两个数 s,t;起点s,终点。n和m为0时输入结束。
(1<n<=1000, 0<m<100000, s != t)
Output
输出 一行有两个数, 最短距离及其花费。
Sample Input
3 2
1 2 5 6
2 3 4 5
1 3
0 0
Sample Output
9 11
思路:和一般的最短路不太一样,多了一个路长相等时比较花费大小的限定,其实都很类似,松弛时路长小就更新,路长一样时花费少也更新,只是优先队列中的结构体重载写法需要注意一下,也算是练习一下优先队列优化dij的好题了。
AC代码:
#include<bits/stdc++.h>
#define INF 0x3F3F3F3F
#define endl '\n'
#define pb push_back
#define css(n) cout<<setiosflags(ios::fixed)<<setprecision(n);
#define sd(a) scanf("%d",&a)
#define sld(a) scanf("%lld",&a)
#define m(a,b) memset(a,b,sizeof a)
#define p_queue priority_queue
using namespace std;
typedef long long ll;
const int maxn=1005;
int n,m;
int s,t;
double a,b;
struct node
{
int to,len,cost;
node(int to=0,int len=0,int cost=0):to(to),len(len),cost(cost){}
};
vector<node> edge[maxn];
int vis[maxn];
int dis[maxn];
int pay[maxn];
struct comp//结构体的优先队列中的比较函数,判断得反着写。
{
bool operator()(const node &a,const node &b)
{
if(a.len==b.len) return a.cost>b.cost;
else return a.len>b.len;
}
};
p_queue<node,vector<node>,comp> pq;//第一个参数是对象类型,第二个是存储结构的载体,一般都用vector,第三个是比较函数。
//priority_queue<node,vector<node>,comp>pq;
dij(int s)
{
memset(dis,0x3f,sizeof(dis));//初始化什么的别忘了。
memset(pay,0x3f,sizeof(pay));
memset(vis,0,sizeof(vis));
dis[s]=0;
pay[s]=0;
pq.push(node(s,0,0));
while(!pq.empty())
{
// int x=pq.top().second;
node head=pq.top();
pq.pop();
int x=head.to;
if(vis[x]) continue;
vis[x]=1;
for(int i=0;i<edge[x].size();i++)
{
int y=edge[x][i].to;
int len=edge[x][i].len;
int cost=edge[x][i].cost;
if(dis[y]>=dis[x]+len)
{
if(dis[y]>dis[x]+len||(dis[y]==dis[x]+len&&pay[y]>pay[x]+cost))
{
dis[y]=dis[x]+len;
pay[y]=pay[x]+cost;
// cout<<pay[y]<<"---"<<endl;
pq.push(node(y,dis[y],pay[y]));
}
}
}
}
}
int main()
{
while(scanf("%d%d",&n,&m)&&n&&m)
{
for(int i=1;i<=n;i++)
{
edge[i].clear();
}
for(int i=0;i<m;i++)
{
int a,b,c,d;
scanf("%d%d%d%d",&a,&b,&c,&d);
node aa;
aa.to=b;
aa.len=c;
aa.cost=d;
edge[a].pb(aa);
aa.to=a;
edge[b].pb(aa);
}
sd(s);sd(t);
dij(s);
printf("%d %d\n",dis[t],pay[t]);
}
return 0;
}