闲来无事学A*。
然后就看到了这个裸的k短路问题。
A*算法我觉得嘛就是三个变量,一个思想。
首先肯定是估价函数h,然后是代价函数g,最后是f=h+g。
思想呢就是每次取出f最小的,然后开始更新周围的节点。
对于未到达的节点直接更新,已到达的节点用g值比较一下,更优则更新。
而估价函数在一开始就算好了。
那么对于k短路问题呢。
首先预处理出终点t到每个点的最短路,即为估价函数h。
然后从起点出发,代价什么的走的时候算一下就好了。
于是当经过一个节点k次的时候就是该节点的k短路了,很朴素的一个启发式搜索。
复杂度什么的我不会算。。。。。
跑起来挺快的倒是
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<cctype>
using namespace std;
const int inf=1e9;
struct Edge{int to,next,v;}g[100005],gt[100005];
int head[1005],headt[1005],cnt,cntt;
void ins(int u,int v,int w){
g[++cnt]=(Edge){v,head[u],w};head[u]=cnt;
gt[++cntt]=(Edge){u,headt[v],w};headt[v]=cntt;
}
struct Dijnode{
int x,d;
Dijnode(int a,int b):x(a),d(b){}
bool operator<(const Dijnode &rhs)const{
return d>rhs.d;
}
};
int dis[1005];
bool done[1005];
void dijkstra(int s){
memset(done,0,sizeof(done));
memset(dis,0x3f,sizeof(dis));
dis[s]=0;
priority_queue<Dijnode>q;
q.push(Dijnode(s,0));
while(!q.empty()){
Dijnode x=q.top();q.pop();
int u=x.x;if(done[u])continue;done[u]=true;
for(int i=headt[u];i;i=gt[i].next){
int v=gt[i].to;
if(dis[v]>dis[u]+gt[i].v){
dis[v]=dis[u]+gt[i].v;
q.push(Dijnode(v,dis[v]));
}
}
}
}
struct Anode{
int f,g,v;
Anode(int a,int b,int c):f(a),g(b),v(c){}
bool operator<(const Anode &rhs)const{
return f>rhs.f;
}
};
int inq[1005];
int A_star(int s,int t,int k){
priority_queue<Anode>q;
if(dis[s]>inf)return -1;
memset(inq,0,sizeof(inq));
q.push(Anode(dis[s],0,s));
while(!q.empty()){
Anode x=q.top();q.pop();
int u=x.v;
inq[u]++;
if(inq[t]==k)return x.f;
if(inq[u]>k)continue;
for(int i=head[u];i;i=g[i].next){
int v=g[i].to,w=g[i].v;
q.push(Anode(dis[v]+w+x.g,x.g+w,v));
}
}
return -1;
}
template<class T>void read(T &x){
static char c;
static bool f;
for(f=0;c=getchar(),!isdigit(c);)if(c=='-')f=1;
for(x=0;isdigit(c);c=getchar())x=x*10+c-'0';
if(f)x=-x;
}
int main(){
int n,m;read(n);read(m);
int u,v,w;
for(int i=1;i<=m;i++){
read(u);read(v);read(w);
ins(u,v,w);
}
int s,t,k;read(s);read(t);read(k);
dijkstra(t);
if(s==t)k++;
printf("%d",A_star(s,t,k));
return 0;
}