CSP2019-03-5 317号子任务

代码是30分

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

#include
#include 
#include 

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>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>x>>y>>z;
        if(zh[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