暴力朴素做法:暴力宽搜,每次取队头,把每次出边加入队,每个点第几次出队就是第几短路
用a_star算法优化:
#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
#include<string.h>
using namespace std;
const int N=2e5+10;
typedef pair<int,int>pll;
typedef pair<int,pll>plll;
int h[N],hr[N],w[N],e[N],ne[N],idx;
int dis[N],f[N],cnt[N];
int s,t,k;
int n,m;
bool st[N];
void add(int *h,int a,int b,int c)
{
e[idx]=b;
w[idx]=c;
ne[idx]=h[a];
h[a]=idx++;
}
void dijistra()
{
priority_queue<pll,vector<pll>,greater<pll> >q;
for(int i=1;i<=n;i++) dis[i]=0x3f3f3f3f;
dis[t]=0;
q.push(make_pair(0,t));
while(q.size())
{
auto t=q.top();
q.pop();
int ver=t.second;
int distance = t.first;
if(st[ver]) continue;
st[ver]=true;
for(int i=hr[ver];~i;i=ne[i])
{
int j=e[i];
if(dis[j]>dis[ver]+w[i]){
dis[j]=dis[ver]+w[i];
q.push(make_pair(dis[j],j));
}
}
}
memcpy(f,dis,sizeof dis);
}
//求第k短路,玩命扩展,迪杰斯特拉求最短路,就扩展一次,需要标记;
int a_star()
{
priority_queue<plll,vector<plll>,greater<plll> >q;
q.push(make_pair(f[s],make_pair(0,s)));
while(q.size())
{
auto tt=q.top();
q.pop();
int ver=tt.second.second,distance=tt.second.first;
if(cnt[ver]>=k) continue;//中途点超过k+1,没必要扩展了
cnt[ver]++;
//到终点;
if(ver==t &&cnt[ver]==k) return distance;//正在试图扩展k+1,等到答案
for(int i=h[ver];~i;i=ne[i])
{
int j=e[i];
if(cnt[j]<k) // 没有扩展k+1次
q.push(make_pair(f[j]+w[i]+distance,make_pair(distance+w[i],j)));
}
}
return -1;
}
int main()
{
memset(h,-1,sizeof h);
memset(hr,-1,sizeof hr);
ios::sync_with_stdio(false);
cin>>n>>m;
for(int i=1;i<=m;i++)
{
int a,b,l;
cin>>a>>b>>l;
add(h,a,b,l);
add(hr,b,a,l);
}
cin>>s>>t>>k;
if(s==t) k++;
dijistra();
cout<<a_star()<<endl;
return 0;
}