码迷,mamicode.com
首页 > 其他好文 > 详细

ccf 201903-5

时间:2020-01-10 22:12:00      阅读:66      评论:0      收藏:0      [点我收藏+]

标签:class   ret   img   style   name   fine   printf   flag   ios   

技术图片

 

 

未优化的迪杰斯特拉:30分的代码

 

#include <stdio.h>
#include <iostream>
#include <queue>
#include <string.h>
#include <algorithm>
#define MAX 10000
#define INF 99999999

using namespace std;

int flag[MAX];
int d[MAX];
int all[MAX][MAX];
bool vis[MAX];
int dis[MAX];
int map[MAX][MAX]; 
int dist[MAX];


void dijkstra(int s,  int count, int n){
    memset(dis,INF,sizeof(dis));
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=n;i++)    dis[i]=map[s][i];
    dis[s]=0;
    int min,k;
    for(int i=1;i<=n;i++){
        min=INF;
        for(int j=1;j<=n;j++){
            if(dis[j]<min && !vis[j]){
                min=dis[j];        k=j;
            }
        }
        vis[k]=1;
        for(int p=1;p<=n;p++){
            if(!vis[p]){
                if(dis[p] > dis[k]+map[k][p])
                    dis[p] = dis[k]+map[k][p];
                }
                
            }
        }
    }
    for(int i=1;i<=n;i++)    all[count][i]=dis[i];
}
int main(){ int n,m,k; int u,v,w; int count; int c,r; while(scanf("%d %d %d",&n,&m,&k)!=EOF){ for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) map[i][j]=INF; memset(flag,0,sizeof(flag)); count=0; for(int i=1;i<=n;i++){ cin >> flag[i]; if(flag[i]) count++; } for(int i=1;i<=m;i++){ scanf("%d %d %d",&u,&v,&w); if(map[u][v] > w) map[u][v] = map[v][u] = w; } c=0; for(int i=1;i<=n;i++){ if(flag[i]){ //求行星据点到每个顶点的距离,不必求任意两点之间的距离 dijkstra(i, c, n); c++; } } for(int i=1;i<=n;i++){ //求n个顶点 memset(d,0,sizeof(d)); for(int j=0;j<count;j++) //count个行星据点,count行 d[j] = all[j][i]; //顶点i到行星据点的距离等于对应的列的值 sort(d,d+count); r=0; for(int p=0;p<k; p++) if(d[p]!=INF) r+=d[p]; printf("%d\n",r); } } return 0; }

ccf 201903-5

标签:class   ret   img   style   name   fine   printf   flag   ios   

原文地址:https://www.cnblogs.com/shiliuxinya/p/12178139.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!