寻找第k短路。看到这道题时,很快就想到了思路,但就是实现不了。看了discuss,可以先用dijkstra求出最短路作为估价函数,在astar来实现求出第k短路,而astar最惯常做法是用优先队列保存节点。
用TPoint来保存astar里的节点,其中to是指向的位置,dis代表当前节点中已经过的距离,all是到目标的总距离。
#include<queue>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define maxn 1005
#define inf 1<<30
int sum1[maxn],sum2[maxn];
int dis[maxn],cnt[maxn];
bool vis[maxn];
int tot1,tot2,n,m,t;
int st,en,kth;
struct TNode
{
int to,next,cost;
bool operator<(const TNode &temp) const
{
return cost>temp.cost;
}
};
struct TPoint
{
int to,dis,all;
bool operator<(const TPoint &temp) const
{
return all>temp.all;
}
};
vector<TNode>ans1,ans2;
priority_queue<TNode>qu;
priority_queue<TPoint>ans;
void addedge(int u,int v,int w,bool flag)
{
TNode tmp;
tmp.to=u,tmp.cost=w;
if(flag)
{
tmp.next=sum1[v],sum1[v]=tot1++;
ans1.push_back(tmp);
}
else
{
tmp.next=sum2[v],sum2[v]=tot2++;
ans2.push_back(tmp);
}
}
void dijkstra()
{
int k;
fill(dis,dis+n+1,inf);
memset(vis,0,sizeof(vis));
while(!qu.empty()) qu.pop();
TNode tmp1,tmp2;
tmp1.to=en,tmp1.cost=0;
qu.push(tmp1);
dis[en]=0;
while(!qu.empty())
{
tmp1=qu.top(),qu.pop();
if(vis[tmp1.to]) continue;
vis[tmp1.to]=1;
for(k=sum1[tmp1.to];k!=-1;k=ans1[k].next)
{
if(dis[ans1[k].to]>dis[tmp1.to]+ans1[k].cost)
{
dis[ans1[k].to]=dis[tmp1.to]+ans1[k].cost;
tmp2.to=ans1[k].to;
tmp2.cost=dis[tmp2.to];
qu.push(tmp2);
}
}
}
}
int astar()
{
int i,j,k;
TPoint tmp1,tmp2;
while(!ans.empty()) ans.pop();
memset(cnt,0,sizeof(cnt));
tmp1.to=st,tmp1.dis=0,tmp1.all=dis[st];
if(tmp1.all==inf) return -1;
ans.push(tmp1);
while(!ans.empty())
{
tmp1=ans.top(),ans.pop();
cnt[tmp1.to]++;
if(cnt[en]==kth) return tmp1.all;
if(cnt[tmp1.to]>kth) continue;
for(k=sum2[tmp1.to];k!=-1;k=ans2[k].next)
{
tmp2.to=ans2[k].to;
tmp2.dis=tmp1.dis+ans2[k].cost;
tmp2.all=tmp2.dis+dis[tmp2.to];
ans.push(tmp2);
}
}
return -1;
}
int main()
{
int i,u,v,w;
while(scanf("%d%d",&n,&m)!=EOF)
{
tot1=tot2=0;
memset(sum1,-1,sizeof(sum1));
memset(sum2,-1,sizeof(sum2));
ans1.clear();
ans2.clear();
for(i=0;i<m;i++)
{
scanf("%d%d%d",&u,&v,&w);
addedge(u,v,w,1);
addedge(v,u,w,0);
}
scanf("%d%d%d",&st,&en,&kth);
dijkstra();
if(st==en) kth++;
printf("%d\n",astar());
}
return 0;
}