POJ 1724
2014-04-30 18:37
204 查看
思路：使用优先队列，每次都取出先前状态中距离最短的状态，注意，入队的不是点也不是边，而是先前满足花费小于K的所有的可行状态，队列里面保存的是先前取的点的状态，这样每次弹出距离最小的状态进行更新，到最后得到的一定是满足距离最小且花费不大于K的解，一旦发现N点在状态中，就可以返回了，因为此时的状态就是最优状态，再弹出其他状态进行更新时结果肯定比此时的大，因为优先队列首先弹出的是最小距离的状态。
#include<iostream>
#include<queue>
#include<cstdio>
#include<cstring>
using namespace std;
#define MAX 105
#define INF 1 << 30
typedef struct{
int to, next, toll, length;
}Node;
typedef struct Status{
int length, toll, u;
friend bool operator < (Status a, Status b){
return a.length > b.length;
}
}Status;
Node edge[MAX*MAX];
int head[MAX];
int K, N, R;
int spfa(int s, int t){
priority_queue<Status>q;
Status node;
node.length = 0;
node.toll = 0;
node.u = s;
q.push(node);
while(!q.empty()){
node = q.top();
q.pop();
if(node.u == t) return node.length;
for(int i = head[node.u];i != -1;i = edge[i].next){
if(edge[i].toll + node.toll <= K){
Status temp;
temp.toll = edge[i].toll + node.toll;
temp.u = edge[i].to;
temp.length = node.length + edge[i].length;
q.push(temp);
}
}
}
return -1;
}
int main(){
int u, v, l, t;
/* freopen("in.c", "r", stdin); */
while(~scanf("%d%d%d", &K, &N, &R)){
memset(head, -1, sizeof(head));
for(int i = 1;i <= R;i ++){
scanf("%d%d%d%d", &u, &v, &l, &t);
edge[i].to = v;
edge[i].length = l;
edge[i].toll = t;
edge[i].next = head[u];
head[u] = i;
}
printf("%d\n", spfa(1, N));
}
return 0;
}
内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理 