UVA 11367 Full Tank? (Dijkstra 变种)

题目链接：https://cn.vjudge.net/problem/UVA-11367

#include<bits/stdc++.h>
//#include<iostream>
//#include<queue>
//#include<cstring>
//#include<cstdio>
using namespace std;

#define debug puts("YES");
#define rep(x,y,z) for(int (x)=(y);(x)<(z);(x)++)
#define read(x,y) scanf("%d%d",&x,&y)
#define ll long long

#define lrt int l,int r,int rt
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define root l,r,rt
const int  maxn =1e4+5;
const int mod=1e9+7;
int INF;
ll powmod(ll x,ll y){ll t; for(t=1;y;y>>=1,x=x*x%mod) if(y&1) t=t*x%mod; return t;}
ll gcd(ll x,ll y){return y?gcd(y,x%y):x;}
/*
题目大意:一张图，每个点都有油价，每条边表示消耗的油量，
然后若干查询，给定油箱容量和始终点，求其中的最短消耗。

在当前选出的点进行扩展的时候，
要枚举在当前点购买的油的总量，把所有点压进去优先队列即可。
因为状态总数是1000*100，所以不会超时，
但注意在队列中如果探测到尾节点，则直接退出，不然会超时。

*/
///每个点每种容量的最优值
int a[maxn],x,y,z;
int dp[maxn][105];
///链式前向星存储边
struct edge{int u,nxt,w;}e[maxn<<1];
int head[maxn],tot=0;
void init(){memset(head,-1,sizeof(head));tot=0;}
void add(int x,int y,int w){e[tot]=edge{y,head[x],w};head[x]=tot++;}
///节点
int c,s,ed;
int n,m,q;
struct node
{
    int x,w,r;///当前节点
    bool operator<(const node& y) const
    {
        return w>y.w;
    }
};
int main()
{
    init();scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++) scanf("%d",&a[i]);
    for(int i=0;i<m;i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z),add(y,x,z);
    }
    scanf("%d",&q);
    for(int i=0;i<q;i++)
    {
        scanf("%d%d%d",&c,&s,&ed);
        memset(dp,0xf,sizeof(dp));INF=dp[0][0];
        priority_queue<node> pq;
        pq.push(node{s,0,0});dp[s][0]=0;
        while(!pq.empty())
        {
            node tp=pq.top();pq.pop();

            int tx=tp.x,tw=tp.r;///当前所剩余的油
            if(tx==ed) break;///不加这句超时

            for(int i=head[tx];~i;i=e[i].nxt)
            {
                int v=e[i].u,w=e[i].w;///权重和扩展节点
                for(int j=0;j<=c-tw;j++)///枚举在当前点要采购多少油
                {
                    if(tw+j-w>=0&&dp[v][tw+j-w]>dp[tx][tw]+j*a[tx])
                    {
                        dp[v][tw+j-w]=dp[tx][tw]+j*a[tx];
                        pq.push(node{v,dp[v][tw+j-w],tw+j-w});
                    }
                }
            }
        }
        int ans=INF;for(int i=0;i<=c;i++) ans=min(ans,dp[ed][i]);
        if(ans==INF) puts("impossible");
        else printf("%d\n",ans);
    }
    return 0;
}
/*
5 5
10 10 20 12 13
0 1 9
0 2 8
1 2 1
1 3 11
2 3 7
2
10 0 3
20 1 4

*/