题解:任取节点2较小邻边的两倍作为mod,若可以通过距离x再次到达节点2,则可以通过x+mod,x+2*mod,x+3*mod...再次到达节点2.
所以利用同余,spfa()跑最短路求d[i][j](i代表从2节点到i节点,j代表从2节点到i节点的距离%mod==j,d[i][j]代表2节点到i节点距离%mod==j的最短距离)
当再次回到2节点时,更新ans,最短距离小于k时,ans=min(ans,tmp+((k-tmp-1)/mod+1)*mod)
每次只扩展当前节点的邻节点i--->(i+1)%4,i--->(i-1+4)%4,所以节点1234分别保存为0123
#include<iostream> #include<queue> #include<stdio.h> #include<string.h> using namespace std; typedef long long ll; const ll inf=1e18+5; const int maxn=60010; struct node{ int p;ll dis; }; int t,e[4][4],mod; ll ans,k,d[4][maxn]; bool vis[4][maxn]; queue<node> q; void spfa(){ q.push(node{1,0});vis[1][0]=1,d[1][0]=0; while(!q.empty()){ int u=q.front().p;ll dis=q.front().dis;vis[u][dis%mod]=0;q.pop(); //printf("%lld\n",ans); for(int i=-1;i<2;i+=2){ int v=(u+i+4)%4; ll tmp=dis+e[u][v]; if(d[v][tmp%mod]>tmp){ d[v][tmp%mod]=tmp; if(!vis[v][tmp%mod]) q.push(node{v,tmp}),vis[v][tmp%mod]=1; if(v==1){ if(tmp>=k) ans=min(ans,tmp); else ans=min(ans,tmp+((k-tmp-1)/mod+1)*mod); } } } } } int main(){ scanf("%d",&t); while(t--){ scanf("%lld",&k); for(int i=0;i<4;i++) for(int j=0;j<maxn;j++) d[i][j]=1e18+10; for(int i=0;i<4;i++)scanf("%d",&e[i][(i+1)%4]),e[(i+1)%4][i]=e[i][(i+1)%4]; mod=2*min(e[0][1],e[1][2]); ans=((k-1)/mod+1)*mod; spfa(); printf("%lld\n",ans); } }