178. 第K短路(bfs)

题目链接:https://www.acwing.com/problem/content/description/180/
给定一张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算法
A
算法核心思想就是预估接下来的情况,达到优先搜索整体最优的目的,以减少搜索层次。
简单的说,就是我们平常用优先队列求最短路的时候,就是拿当前最少的距离来拓展,现在就要考虑一下他之后的情况,在拓展了。

1,我们可以预处理一下以T为起点到其他点的最短路径(注意,这是个有向图。现在是求T到其他的点,那么我们需要新建一个和给定图相反方向的图。555 一开始竟然没考虑到)。
2,优先队列,不过里面考虑比较的值为实际距离长度+预估长度。

#include"stdio.h"
#include"string.h"
#include"vector"
#include"queue"
#include"algorithm"
using namespace std;
typedef pair<int,int> PII;
#define INF 100101010

typedef struct Node
{
    int u,real,f;
    friend bool operator < (Node a,Node b)
    {
        return a.f > b.f;
    }
    Node(){}
    Node(int a,int b,int c)
    {
        u = a; real = b; f = c;
    }
}Node;

int num_cnt[1010];
int dist_min[1010];
vector<PII> Graph[1010];
vector<PII> G[1010];
int S,T,N,M,K;
int minx_K;

struct cmp
{
    bool operator()(const PII p1,const PII p2)
    {
        return p1.second > p2.second;
    }
};
void bfs_min()
{
    priority_queue<PII,vector<PII>,cmp > Q;
    Q.push({T,0});
    int vis[1010];
    memset(vis,0,sizeof(vis));
    dist_min[T] = 0;
    while(!Q.empty())
    {
        PII T = Q.top();
        Q.pop();
        if(vis[T.first] == 0)
        {
            for(int i = 0; i < G[T.first].size(); i ++)
            {
                PII t = G[T.first][i];
                if(dist_min[t.first] > dist_min[T.first] + t.second)
                {
                    dist_min[t.first] = dist_min[T.first] + t.second;
                    Q.push({t.first, dist_min[t.first]});
                }
            }
            vis[T.first] = 1;
        }
    }
}

int bfs_k()
{
    priority_queue<Node> Q;
    Q.push({S,0,0 + dist_min[S]});
    while(!Q.empty())
    {
        Node A = Q.top(); Q.pop();
        num_cnt[A.u] ++;
        if(num_cnt[A.u] == K && A.u == T)
        {
            minx_K = A.real; return 1;
        }
        for(int i = 0; i < Graph[A.u].size(); i ++)
        {
            PII X = Graph[A.u][i];
           // if(num_cnt[X.first] > K) continue;
            Q.push({X.first,A.real + X.second,A.real + X.second + dist_min[X.first]});
        }
    }
    return -1;
}

int main()
{
    scanf("%d%d",&N,&M);
    for(int i = 0; i <= N; i ++) dist_min[i] = INF;
    for(int i = 1; i <= M; i ++)
    {
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        Graph[u].push_back({v,w});
        G[v].push_back({u,w});
    }
    scanf("%d%d%d",&S,&T,&K);
    if(S == T) K ++;
    bfs_min();

    if(dist_min[S] == INF || M == 0)
    {
        printf("-1\n"); return 0;
    }
    if(bfs_k() == 1)
    printf("%d\n",minx_K);
    else
    printf("-1\n");
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值