第16次CCF CSP认证-第5题-317 号子任务(subtask317)-图论最短路径

【题目背景】
“你在平原上走着走着，突然迎面遇到一堵墙，这墙向上无限高，向下无限深，向
左无限远，向右无限远，这墙是什么？”——《流浪地球》原著
我们带着地球去流浪了，为了处理流浪过程中可能会发生的危机，联合政府找到
你，希望你能协助完成 317 号子任务：制定应急预案。
【题目描述】
地球的表面有 n 个据点，这些据点之间存在 m 条双向道路。
这些据点中，有的是建立在行星发动机之下，受到行星发动机的保护（行星发动机
据点），而其他据点则没有行星发动机的保护（普通据点，比如燃料采集据点/科研据点
等）。
当发生危机的时候，没有行星发动机的保护是非常危险的，所以每个人都需要赶到
最近的行星发动机据点寻求庇护，然而行星发动机据点也不一定安全，再加上行星发动
机据点容量有限，所以有些时候得去第二近或者第三近的行星发动机据点。
联合政府找到你，希望你能够计算出每个据点最近的 k 个行星发动机据点，为了
简化问题，你只需要输出每个据点到最近 k 个行星发动机据点的最短距离之和，如果
某个据点能够到达的行星发动机据点不足 k 个，则输出其能到达的所有行星发动机的
最短距离之和。
【输入格式】
从标准输入读入数据。
输入的第一行包含三个用空格隔开的整数 n, m, k ，含义见题目᧿述，保证 1 ≤ n ≤
104, 0 ≤ m ≤ 104, 1 ≤ k ≤ 102。据点依次编号为 1 到 n 。
第二行包含 n 个整数依次表示每个据点的类型，每个数为 1 或 0 （1 表示对应据
点为行星发动机据点，0 表示普通据点）。
接下来 m 行，每行三个整数 u, v,w 表示有一条长度为 w 的双向道路连接 u 号据点
和 v 号据点，1 ≤ u, v ≤ n, 1 ≤ w ≤ 103 。
可能有重边和自环。
【输出格式】
输出到标准输出。
输出 n 行，每行输出一个整数表示答案（见题目᧿述）

【样例 1 输入】
7 6 2
1 0 1 0 1 1 0
1 4 1
1 2 3
2 4 4
2 3 5
2 5 7
6 7 5
【样例 1 输出】

8

8

10

10

0

5

【样例 1 解释】
该样例的输入对应的图如下，其中红色点是行星发动机据点，白色点是普通据点。

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #define ll long long
 using namespace std;
 ;
 const int inf=9e7;
 ll ans[amn];
 int typ[amn],e[amn][amn],k,n,m;
 int main()
 {
     cin>>n>>m>>k;
     ; i<=n; i++)
     {
         ; j<=n; j++)
         {
             ;
             else e[i][j]=inf;
         }
     }
     ; i<=n; i++)
     {
         int tt;
         cin>>tt;
         typ[i]=tt;
     }
     int u,v,w;
     ; i<=m; i++)
     {
         cin>>u>>v>>w;
         e[u][v]=w;
         e[v][u]=w;
     }
     ; c<=n; c++)
     {
         ; i<=n; i++)
         {
             ; j<=n; j++)
                 if(e[i][k]<inf&&e[k][j]<inf&&e[i][j]>e[i][c]+e[c][j])
                     e[i][j]=e[i][c]+e[c][j];
         }
     }
 //    cout<<endl<<endl;///查看图的最短路径(调试用)
 //    for(int i=1; i<=n; i++)
 //    {
 //        for(int j=1; j<=n; j++)
 //        {
 //            printf("%10d",e[i][j]);
 //        }
 //        cout<<endl<<endl;
 //    }
     ; i<=n; i++)
     {
         ;
         memset(ans,,sizeof(ans));
         ; j<=n; j++)
         {
             if(e[i][j]<inf&&typ[j])
             {
                 ans[top++]=e[i][j];
             }
         }
         sort(ans,ans+top);
         ll sum=;
         ; i<k&&i<top; i++)
         {
             sum+=ans[i];
         }
         cout<<sum<<endl;
     }
     ;
 }
 /*
 样例

 7 6 2
 1 0 1 0 1 1 0
 1 4 1
 1 2 3
 2 4 4
 2 3 5
 2 5 7
 6 7 5

 */