CCF 201903-5 317号子任务 30分

最新推荐文章于 2023-04-19 17:59:42 发布

原创最新推荐文章于 2023-04-19 17:59:42 发布 · 323 阅读

0 ·

CC 4.0 BY-SA版权

CCF 专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一个基于Floyd算法和优先队列实现的最短路径查找程序，该程序能够处理复杂的图结构，找出从任意节点到被标记为Is_im的节点的最短路径，并计算前K个最短距离的总和。通过使用C++标准库中的优先队列，程序实现了高效的路径搜索。

部署运行你感兴趣的模型镜像

#include<bits/stdc++.h>
using namespace std;
const int maxn = 10010;
const int INF = 0x3f3f3f3f;
bool Is_im[maxn];
int Map[maxn][maxn];

int main()
{
    int n, m, K;
    scanf("%d %d %d", &n, &m, &K);
    for(int i = 1; i <= n; i++)
       scanf("%d", &Is_im[i]);
    memset(Map, 0x3f, sizeof(Map));
    for(int i = 1; i <= n; i++)
        Map[i][i] = 0;
    for(int i = 0; i < m; i++){
        int v1, v2, c;
        scanf("%d %d %d", &v1, &v2, &c);
        Map[v1][v2] = Map[v2][v1] = min(Map[v1][v2], c);
    }
    for(int k = 1; k <= n; k++){
        for(int i = 1; i <= n; i++){
            for(int j = 1; j <= n; j++)
                if(Map[i][j] > Map[i][k]+Map[k][j])
                Map[i][j] = Map[i][k] + Map[k][j];
        }
    }
    priority_queue<int, vector<int>, greater<int> > p_q;
    for(int i = 1; i <= n; i++){
        for(int j = 1; j <= n; j++){
            if(Is_im[j] && Map[i][j] < INF){
                p_q.push(Map[i][j]);
            }
        }
        int KK = 0;
        long long sum = 0;
        while(!p_q.empty()){
            int num = p_q.top();
            p_q.pop();
            sum += num;
            KK++;
            if(KK == K) break;
        }
        printf("%lld\n", sum);
        while(!p_q.empty())
            p_q.pop();

    }
    return 0;
}

您可能感兴趣的与本文相关的镜像