POJ1724(ROADS)

题目链接

有限制的单源最短路。

题目大意：给定一个图，图中的每条边有一个长度和一个费用，给出最大费用，求在不超过最大费用的前提下的最短路（从s到e）。

我的解法是BFS+优先队列，我们可以把<d,t,i>看成状态，i为结点编号，d为结点到源点的距离，t为目前剩余的费用，每次BFS时都选取d最小的状态进行扩展，直到到达目标结点。一开始，我还担心内存会爆掉，但是一下也没想到其他方法，就照这个想法写了，结果居然AC了。

由于要用到优先队列，所以只好临时学了一点STL，第一次提交是忘改用c++提交，CE了一次。

#include <stdio.h>
#include <memory.h>
#include <queue>
#define N 101
#define M 10001

using namespace std;
typedef pair<pair<int,int>,int> node;
priority_queue<node,vector<node>,greater<node> > pq;

int n,m;
int u[M],v[M],d[M],t[M],next[M];
int first[N];

int main()
{
  int i,j,k,e,di,ti,ans;
  node tmp;
  while(~scanf("%d%d%d",&k,&n,&m))
  {
    for(e=1;e<=m;e++)
    {
      scanf("%d%d%d%d",&u[e],&v[e],&d[e],&t[e]);
      next[e]=first[u[e]];
      first[u[e]]=e;
    }
    pq.push(make_pair(make_pair(0,k),1));
    ans=-1;
    while(!pq.empty())
    {
      tmp=pq.top(),pq.pop();
      if(ans!=-1) continue;
      i=tmp.second;
      di=tmp.first.first;
      ti=tmp.first.second;
      if(i==n)  ans=di;
      for(e=first[i];e;e=next[e])
      {
        j=v[e];
        if(ti>=t[e])  pq.push(make_pair(make_pair(di+d[e],ti-t[e]),j));
      }
    }
    printf("%d\n",ans);
    memset(first,0,sizeof(first));
  }
  return 0;
}