K短路的裸题。。
dij:
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<vector>
#include<ctime>
#include<cstdlib>
#include<queue>
#include<ext/pb_ds/priority_queue.hpp>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define pa pair<int,int>
#define fd(i,a,b) for(int i=a;i>=b;i--)
#define inf 2000000000
using namespace std;
using namespace __gnu_pbds;
const int N=1e5+5;
typedef __gnu_pbds::priority_queue<pa,greater<pa> > heap;
int n,m,K,cnt,cnt2;
int ans[N],d[N],g[N],tim[N];
int last[N],last2[N];
heap q;
heap::point_iterator id[N];
struct edge
{
int to,next,v;
}e[N],ed[N];
inline void insert(int x,int y,int z)
{
e[++cnt].to=y;e[cnt].next=last[x];e[cnt].v=z;last[x]=cnt;
ed[++cnt2].to=x;ed[cnt2].next=last2[y];last2[y]=cnt2;ed[cnt2].v=z;
}
inline void dij()
{
fo(i,1,n)d[i]=inf;
d[1]=0;
id[1]=q.push(make_pair(0,1));
while(!q.empty())
{
int now=q.top().second;q.pop();
for(int i=last2[now];i;i=ed[i].next)
if (ed[i].v+d[now]<d[ed[i].to])
{
d[ed[i].to]=d[now]+ed[i].v;
if (id[ed[i].to]!=0)
q.modify(id[ed[i].to],make_pair(d[ed[i].to],ed[i].to));
else id[ed[i].to]=q.push(make_pair(d[ed[i].to],ed[i].to));
}
}
}
inline void Astar()
{
if (d[n]==inf)return;
q.push(make_pair(d[n],n));
while (!q.empty())
{
int now=q.top().second;
int dis=q.top().first;
q.pop();
tim[now]++;
if (now==1)ans[tim[now]]=dis;
if (tim[now]<=K)
for(int i=last[now];i;i=e[i].next)
q.push(make_pair(dis-d[now]+d[e[i].to]+e[i].v,e[i].to));
}
}
int main()
{
scanf("%d%d%d",&n,&m,&K);
fo(i,1,m)
{
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
insert(x,y,z);
}
dij();
Astar();
fo(i,1,K)
if (ans[i])printf("%d\n",ans[i]);
else puts("-1");
return 0;
}
spfa(来自zyf2000。。稍后我自己打一份。。)
呼呼。。终于打完了。。
cnt表示的是到这个点的路径数量,估价函数的话自己看看相关教程吧。
具体做法基本上就是先跑一边spfa,求出Astar中的f,然后求h和g的话Astar就好了。
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=2e5+5;
int n,m;
typedef long long ll;
ll dis[N],ans[N],val[N],val1[N];
int cnt[N];
int tot;
int s,t,k;
int head[N],next[N],go[N];
int tot1,head1[N],next1[N],go1[N];
struct node
{
int x;
ll g,h;
bool operator < (const node &a) const
{
return g+h>a.g+a.h;
}
};
inline void add(int x,int y,ll z)
{
go[++tot]=y;next[tot]=head[x];val[tot]=z;head[x]=tot;
go1[++tot1]=x;next1[tot1]=head1[y];val1[tot1]=z;head1[y]=tot1;
}
bool vis[N];
int q[N];
inline void spfa()
{
memset(dis,127,sizeof(dis));
dis[t]=0;
vis[t]=1;
q[1]=t;
int t=0,w=1;
while (t<w)
{
int x=q[++t];
for(int i=head1[x];i;i=next1[i])
{
int v=go1[i];
if (dis[v]>dis[x]+val1[i])
{
dis[v]=dis[x]+val1[i];
if (!vis[v])vis[v]=1,q[++w]=v;
}
}
vis[x]=0;
}
}
inline void Astar()
{
priority_queue <node> q;
q.push((node){s,0,dis[s]});
int size=0;
while (!q.empty())
{
node x=q.top();
q.pop();
++cnt[x.x];
if (cnt[x.x]>k)continue;
if (x.x==t)ans[++size]=x.g;
if (cnt[t]==k)return;
for(int i=head[x.x];i;i=next[i])
q.push((node){go[i],x.g+val[i],dis[go[i]]});
}
}
int main()
{
scanf("%d%d%d",&n,&m,&k);
fo(i,1,m)
{
int x,y;
ll z;
scanf("%d%d%lld",&x,&y,&z);
if (x<y)swap(x,y);
add(x,y,z);
}
s=n,t=1;
spfa();
memset(ans,-1,sizeof(ans));
Astar();
fo(i,1,k)printf("%lld\n",ans[i]);
}