1. 程式人生 > >Convenient Location Aizu

Convenient Location Aizu

傳送門

題意:大概意思是說,選定一個地點使得從該地點開始到達剩餘每個地點的總長度加起來最小,找出這個點並且找出這個點所到達其他點的總長度,最後輸出。

題解:兩種解法:一,既可以使用spfa解法,對每個點跑出所有的最短路並進行更新,最後得出答案

附上程式碼:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<queue>

using namespace std;

const int maxn=50;
const int inf=0x3f3f3f3f;

struct edge{
    int v,cost;
    edge(int _v,int _cost):v(_v),cost(_cost){}
};

vector<edge>ve[maxn];

bool vis[maxn];
int dist[maxn];

int n,N,MIN,pos;

void init()
{
    pos=-1;
    MIN=inf;
    for(int i=0;i<=n;i++){
        ve[i].clear();
    }
    int temp=0;
    for(int i=1;i<=n;i++){
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        ve[a].push_back(edge(b,c));
        ve[b].push_back(edge(a,c));
        temp=max(temp,a);
        temp=max(temp,b);
    }
    N=temp;
}

void spfa(int start)
{
    memset(vis,0,sizeof(vis));
    for(int i=0;i<=N;i++){
        dist[i]=inf;
    }
    vis[start]=true;
    dist[start]=0;
    queue<int>que;
    while(!que.empty()){
        que.pop();
    }
    que.push(start);
    while(!que.empty()){
        int u=que.front();
        que.pop();
        vis[u]=false;
        for(int i=0;i<ve[u].size();i++){
            int v=ve[u][i].v;
            if(dist[v]>dist[u]+ve[u][i].cost){
                dist[v]=dist[u]+ve[u][i].cost;
                if(!vis[v]){
                    vis[v]=true;
                    que.push(v);
                }
            }
        }
    }
}

bool update_min()
{
    int sum=0;
    for(int i=0;i<=N;i++){
        sum+=dist[i];
    }
    if(sum<MIN){
        MIN=sum;
        return true;
    }
    return false;
}

int main()
{
    while(scanf("%d",&n)&&n){
        init();
        for(int i=0;i<=N;i++){
            spfa(i);
            if(update_min()){
                pos=i;
            }
        }
        printf("%d %d\n",pos,MIN);
    }
    return 0;
}

解法二:使用floyd解法,資料量比較小

#include<iostream>
#include<cstdio>

using namespace std;

const int inf=0x3f3f3f3f;

int graph[15][15];

void init(int n)
{
   for(int i=0;i<n;i++){
        for(int j=0;j<i;j++){
            graph[i][j]=graph[j][i]=inf;
        }
        graph[i][i]=0;
    }
}

void floyd(int n)
{
    for(int k=0;k<n;k++){
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                graph[i][j]=min(graph[i][j],graph[i][k]+graph[k][j]);
            }
        }
    }
}


int main()
{
    int n;
    while(scanf("%d",&n)&&n){
        init(15);
        int m=0;
        for(int i=0;i<n;i++){
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            graph[a][b]=graph[b][a]=min(graph[a][b],c);
            m=max(max(m,a),b);
        }
        floyd(m+1);
        int ans=inf,kase=0;
        for(int i=0;i<=m;i++){
            int tp=0;
            for(int j=0;j<=m;j++){
                if(graph[i][j]!=inf){
                    tp+=graph[i][j];
                }
            }
            if(tp<ans){
                ans=tp;
                kase=i;
            }
        }
        printf("%d %d\n",kase,ans);
    }
    return 0;
}