POJ3635 Full Tank?【Dijkstra+DP】

题意：

n个城市之间有m条双向路。每条路要耗费一定的油量。每个城市的油价是固定并且已经给出的。有q个询问，表示从城市s走到e，油箱的容量为c，求最便宜的方案。

思路：

用Dijkstra+Heap即可求有环的动态规划.

代码：

#include <cstdio>
#include <cstring>
#include <queue>

using namespace std;

const int N = , M = , inf = 0x7f7f7f7f;
int h[N], p[M], v[M], w[M], c[N], cnt = ;
int dp[N][], vis[N][], s, t, cap;

struct Data
{
    int x, o, c;
};

bool operator <(Data a, Data b)
{
    return a.c>b.c;
}

void add(int x, int y, int z)
{
    p[++cnt] = h[x]; w[cnt] = z; v[cnt] = y; h[x] = cnt;
}

int dij()
{
    priority_queue<Data> q;
    memset(dp, , sizeof dp);
    memset(vis, , sizeof vis);
    dp[s][] = ;
    q.push((Data) { s,  });
    while (!q.empty())
    {
        Data u = q.top();
        q.pop();
        vis[u.x][u.o] = ;
        if (u.x == t) return u.c;
        if (u.o < cap && !vis[u.x][u.o + ] &&
            dp[u.x][u.o + ] > dp[u.x][u.o] + c[u.x])
            {
            dp[u.x][u.o + ] = dp[u.x][u.o] + c[u.x];
            q.push((Data) {u.x, u.o + , dp[u.x][u.o + ]});
            }
        for (int i = h[u.x]; i; i = p[i])
            if (u.o >= w[i] && !vis[v[i]][u.o - w[i]] && dp[v[i]][u.o - w[i]] > dp[u.x][u.o])
                {
                    dp[v[i]][u.o - w[i]] = dp[u.x][u.o];
                    q.push((Data) {v[i], u.o - w[i], dp[v[i]][u.o - w[i]]});
                }
    }
    return -;
}

int main()
{
    int n, i, m, q;
    scanf("%d%d", &n, &m);
    for(i=;i<n;i++)
        scanf("%d", c + i);
    while (m--)
    {
        scanf("%d%d%d", &s, &t, &cap);
        add(s, t, cap);
        add(t, s, cap);
    }
    scanf("%d", &q);
    while (q--)
    {
        scanf("%d%d%d", &cap, &s, &t);
        int re = dij();
        if (re == -) puts("impossible");
        else printf("%d\n", re);
    }
    return ;
}