给定一张N个点(编号1,2…N),M条边的有向图,求从起点S到终点T的第K短路的长度,路径允许重复经过点或边。
注意: 每条最短路中至少要包含一条边。
输入格式
第一行包含两个整数N和M。
接下来M行,每行包含三个整数A,B和L,表示点A与点B之间存在有向边,且边长为L。
最后一行包含三个整数S,T和K,分别表示起点S,终点T和第K短路。
输出格式
输出占一行,包含一个整数,表示第K短路的长度,如果第K短路不存在,则输出“-1”。
数据范围
1≤S,T≤N≤1000,
0≤M≤105,
1≤K≤1000,
1≤L≤100
输入样例:
2 2
1 2 5
2 1 4
1 2 2
输出样例:
14
题意:如题 |
思路:
A*优雅的暴力算法,还不是特别清楚过程,先记录一下。
AC代码:
#include<iostream>
#include<string>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include <queue>
#include<sstream>
#include <stack>
#include <set>
#include <bitset>
#include<vector>
#define FAST ios::sync_with_stdio(false)
#define abs(a) ((a)>=0?(a):-(a))
#define sz(x) ((int)(x).size())
#define all(x) (x).begin(),(x).end()
#define mem(a,b) memset(a,b,sizeof(a))
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define rep(i,a,n) for(int i=a;i<=n;++i)
#define per(i,n,a) for(int i=n;i>=a;--i)
#define pb push_back
#define mp make_pair
#define fi first
#define se second
using namespace std;
typedef long long ll;
typedef pair<ll,ll> PII;
typedef pair<ll,PII> PIII;
const int maxn = 1e5+200;
const int inf=0x3f3f3f3f;
const double eps = 1e-7;
const double pi=acos(-1.0);
const int mod = 1e9+7;
inline int lowbit(int x){return x&(-x);}
ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
void ex_gcd(ll a,ll b,ll &d,ll &x,ll &y){if(!b){d=a,x=1,y=0;}else{ex_gcd(b,a%b,d,y,x);y-=x*(a/b);}}//x=(x%(b/d)+(b/d))%(b/d);
inline ll qpow(ll a,ll b,ll MOD=mod){ll res=1;a%=MOD;while(b>0){if(b&1)res=res*a%MOD;a=a*a%MOD;b>>=1;}return res;}
inline ll inv(ll x,ll p){return qpow(x,p-2,p);}
inline ll Jos(ll n,ll k,ll s=1){ll res=0;rep(i,1,n+1) res=(res+k)%i;return (res+s)%n;}
inline ll read(){ ll f = 1; ll x = 0;char ch = getchar();while(ch>'9'|ch<'0') {if(ch=='-') f=-1; ch = getchar();}while(ch>='0'&&ch<='9') x = (x<<3) + (x<<1) + ch - '0', ch = getchar();return x*f; }
int dir[4][2] = { {1,0}, {-1,0},{0,1},{0,-1} };
const int V = 2e3, E=2e5+500;
ll d[V],cost[E];
ll n,m,s,t;
ll k;
ll head[V],tail[V], pnt[E],nxt[E],e=1; //head引导正向边,tail引导反向边
ll f[V];
ll vis[V], sta[V];
void addedge(ll *head, ll u,ll v,ll c)
{
pnt[e]=v; //当前以u为顶点,c为边长的,到v的一条边
cost[e]=c; //存入当前边权值
nxt[e]=head[u]; //下一个其实是前一个
head[u]=e++; //当前边编号
}
void Dijkstra() //先用一次dijkstra求出t到各个点的最短路,作为估计值
{
rep(i,0,n+1) d[i] = inf;
priority_queue<PII, vector<PII>, greater<PII> > q;
d[t] = 0;
q.push(mp(0LL,t));
while(!q.empty())
{
ll x = q.top().se; q.pop();
if(vis[x]) continue;
vis[x] = 1;
for(int i=tail[x]; i ;i = nxt[i])
{
ll v = pnt[i];
if(d[v]>d[x]+cost[i]&&!vis[v])
{
d[v] = d[x] + cost[i];
q.push(mp(d[v], v));
}
}
}
memcpy(f,d,sizeof(d)); //丢给f数组
}
ll A_star()
{
priority_queue<PIII, vector<PIII>, greater<PIII> > q; //pair< 估计值+ 到当前最短路,pair<到当前最短路,结点> >
q.push(mp(f[s], mp(0LL,s) ) ); //初始
while(!q.empty())
{
ll x = q.top().se.se; ll dis = q.top().se.fi; q.pop();
if(sta[x]>=k) continue; //对于某个点如果遇到了k次以上,就不用考虑了。
sta[x] ++; //计数++
if(x==t&&sta[x]==k) return dis; //到终点,且为第k次出队,就是答案了。
for(int i=head[x];i ;i = nxt[i])
{
ll v = pnt[i];
if(sta[v]<k) //还没到k次就进队
{
ll tmp = dis + cost[i];
q.push(mp(tmp + f[v], mp(tmp,v) ));
}
}
}
return -1;
}
int main()
{
n = read(), m = read();
rep(i,1,m)
{
ll x = read(), y = read(), z = read();
addedge(head,x,y,z), addedge(tail,y,x,z); //正反边
}
s = read() , t = read() , k = read();
if(s==t) k++; //起点终点重合的时候k++确保有解的时候不会错
Dijkstra();
cout<<A_star()<<'\n';
return 0;
}