luogu P5340 [TJOI2019]大中锋的游乐场
要求经过路径汉堡的点和可乐的点个数之差绝对值\(\le k\),所以可以考虑dp,\(f_{i,j}\)表示到点\(i\),汉堡的点个数减可乐的点的个数为\(j\)的最短距离,注意一下负下标处理,然后跑个dij就完事了
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<vector>
#include<cmath>
#include<ctime>
#include<queue>
#include<map>
#include<set>
#define LL long long
#define db double
using namespace std;
const int N=1e4+10,M=1e5+10,mod=998244353;
int rd()
{
int x=0,w=1;char ch=0;
while(ch<'0'||ch>'9'){if(ch=='-') w=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+(ch^48);ch=getchar();}
return x*w;
}
int to[M<<1],nt[M<<1],w[M<<1],hd[N],tot;
void add(int x,int y,int z)
{
++tot,to[tot]=y,nt[tot]=hd[x],w[tot]=z,hd[x]=tot;
++tot,to[tot]=x,nt[tot]=hd[y],w[tot]=z,hd[y]=tot;
}
int n,m,kk,a[N];
LL di[N][21];
struct node
{
int x,k;
LL d;
bool operator < (const node &bb) const {return d>bb.d;}
};
priority_queue<node> q;
int main()
{
int T=rd();
while(T--)
{
n=rd(),m=rd(),kk=rd();
for(int i=1;i<=n;++i) a[i]=rd()&1?-1:1;
memset(hd,0,sizeof(hd)),tot=1;
while(m--)
{
int x=rd(),y=rd(),z=rd();
add(x,y,z);
}
memset(di,0x3f,sizeof(di));
int ps=rd(),pt=rd();
if(kk+a[ps]>=0&&kk+a[ps]<=kk+kk)
q.push((node){ps,kk+a[ps],di[ps][kk+a[ps]]=0});
while(!q.empty())
{
int x=q.top().x,k=q.top().k;
LL d=q.top().d;
q.pop();
if(d>di[x][k]) continue;
for(int i=hd[x];i;i=nt[i])
{
int y=to[i],nk=k+a[y];
if(nk>=0&&nk<=kk+kk&&di[y][nk]>di[x][k]+w[i])
q.push((node){y,nk,di[y][nk]=di[x][k]+w[i]});
}
}
LL ans=1ll<<50;
for(int i=0;i<=kk+kk;++i) ans=min(ans,di[pt][i]);
ans<(1ll<<50)?printf("%lld\n",ans):puts("-1");
}
return 0;
}