CSP2019-03-5 317号子任务

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

Q_000

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

阅读量456

点赞数 1

分类专栏： CSP

本文链接：https://blog.csdn.net/Qzy_000/article/details/99962042

版权

CSP 专栏收录该内容

11 篇文章 0 订阅

订阅专栏

代码是30分

采用的方法是邻接矩阵和Floyd算法，第一次写的比较复杂，超了内存限制，修改后超时只能得30分，简单搜了一下没找到满分代码，或许用邻接表能过？不过不打算改了，

#include<iostream>
#include <cstring>
#include <algorithm>

using namespace std;
#define INF 0x3f3f3f3f

typedef struct nodee
{
    int a[10005];
}node;
node h[10005];
int b[10005];
int temp[10005];
int main()
{
    int n,m,k;
    cin>>n>>m>>k;
    for(int i = 0;i<n;i++)
    cin>>b[i+1];
    for(int i = 0;i<=n;i++)
    for(int j = 0;j<=n;j++)
    {
        h[i].a[j] = INF;
        if(i==j)h[i].a[j] = 0;
        
    }
    
    for(int i = 0;i<m;i++)
    {
        int x,y,z;
        cin>>x>>y>>z;
        if(z<h[x].a[y])
        {
            h[x].a[y] = z;
            h[y].a[x] = z;
        }
    }
    
    
    for(int k = 1;k<=n;k++)
    for(int i = 1;i<=n;i++)
    for(int j = i+1;j<=n;j++)
    if(h[i].a[j]>h[i].a[k]+h[k].a[j])
    {
        h[i].a[j]=h[i].a[k]+h[k].a[j];
        h[j].a[i]=h[i].a[k]+h[k].a[j];
    }
    //cout<<1;
    
    
    for(int i =1;i<=n;i++)
    {
        memset(temp,0,sizeof(temp));
        int count=0;
        for(int j = 1;j<=n;j++)
        {
            if (h[i].a[j] < INF&&b[j]==1)
                temp[count++] = h[i].a[j];     
        }
        sort(temp,temp+count);
        long long  sum = 0;
        for(int j=0;j<k&&j<count;j++)sum+=temp[j];
        cout<<sum<<endl;
    }
    return 0;

}