CCF-CSP 201903-5 317号子任务【Floyd】（30分）

试题编号： 201903-5
试题名称： 317号子任务
时间限制： 1.0s
内存限制： 512.0MB
题目描述：



只会Floyd拿前30分，能拿60分的算法不会写，能拿100分的算法至今还没听说

代码如下（30分）：

#include 
#include 
#include 
using namespace std;
#define INF 0x3f3f3f3f
static const int MAXN=10000;
int E[MAXN+10][MAXN+10],kind[MAXN+10];
struct Node{
    int k,e;
}t[MAXN+10];
bool cmp(Node P,Node Q)
{
    if(P.k==Q.k)
        return P.e<Q.e;
    return P.k>Q.k;
}
int main()
{
    std::ios::sync_with_stdio(false);
    std::cin.tie(0),cout.tie(0);
    int n,m,k;
    cin>>n>>m>>k;
    for(int i=1;i<=n;i++)
        cin>>kind[i];
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            if(i==j)
                E[i][j]=0;
            else
                E[i][j]=INF;  //初始化
    int a,b,c;
    for(int i=1;i<=m;i++)
    {
        cin>>a>>b>>c;
        E[a][b]=min(E[a][b],c); //有重边
        E[b][a]=E[a][b];
    }
    for(int t=1;t<=n;t++)
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                E[i][j]=min(E[i][j],E[i][t]+E[t][j]);  //Floyd求多源最短路径
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            t[j].e=E[i][j];
            t[j].k=kind[j];
        }
        sort(t+1,t+n+1,cmp);
        int cnt=1,sum=0;
        while(cnt<=k)
        {
            if(t[cnt].k && t[cnt].e!=INF)
                sum+=t[cnt].e;
            else
                break;
            cnt++;
        }
        cout<<sum<<endl;
    }
    return 0;
}