题解Luogu P5340 [TJOI2019]大中锋的游乐场

Posted 2021-12-10 yzhang-rp-inf

原题传送门

没想到省选也会出这种题？？！

实际就是一个带有限制的最短路

因为\(k<=10\)，所以我们珂以暴力将每个点的权值分为[-k,k]，为了方便我们珂以转化成[0,2k]，将汉堡的权值记为1，可乐的权值记为-1，最短路即可，如果发现不合理的就果断扔掉即可（不知道有没有好事之徒用SPFA写）

#include <bits/stdc++.h>
#define N 10005
#define pi pair<int,int>
#define getchar nc
using namespace std;
inline char nc(){
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int read()
{
    register int x=0,f=1;register char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
    return x*f;
}
inline void write(register int x)
{
    if(!x)putchar('0');if(x<0)x=-x,putchar('-');
    static int sta[20];register int tot=0;
    while(x)sta[tot++]=x%10,x/=10;
    while(tot)putchar(sta[--tot]+48);
}
inline int Min(register int a,register int b)
{
    return a<b?a:b;
}
int T,n,m,k,c[N],s,t,dis[N][21],vis[N][21];
vector<pi> e[N];
priority_queue<pi,vector<pi>,greater<pi> > Q;
int main()
{
    T=read();
    while(T--)
    {
        n=read(),m=read(),k=read();
        for(register int i=0;i<n;++i)
            e[i].clear(),c[i]=(read()<<1)-3;
        for(register int i=0;i<m;++i)
        {
            int x=read()-1,y=read()-1,z=read();
            e[x].push_back(make_pair(y,z));
            e[y].push_back(make_pair(x,z));
        }
        s=read()-1,t=read()-1;
        memset(dis,63,sizeof(dis));
        memset(vis,0,sizeof(vis));
        Q.push(make_pair(dis[s][k+c[s]]=0,n*(k+c[s])+s));
        while(!Q.empty())
        {
            int u=Q.top().second%n,x=Q.top().second/n;
            Q.pop();
            if(vis[u][x])
                continue;
            vis[u][x]=1;
            for(register int t=0;t<e[u].size();++t)
            {
                int v=e[u][t].first,w=e[u][t].second,y=x+c[v];
                if(y>=0&&y<=k<<1&&dis[v][y]>dis[u][x]+w)
                    dis[v][y]=dis[u][x]+w,Q.push(make_pair(dis[v][y],n*y+v));
            }
        }
        int ans=dis[t][0];
        for(register int i=1;i<=k<<1;++i)
            ans=Min(ans,dis[t][i]);
        write(ans>1e9?-1:ans),puts("");
    }
    return 0;
}