寝癖頭の解法

学習中の覚え書きを投稿、更新していきます。

Aizu Online Judge in C++ #ALDS1_12_A : Minimum Spanning Tree

Aizu Online Judge(AOJ)の過去問から、その提出コードの解答例です。

・問題 "Minimum Spanning Tree"
https://onlinejudge.u-aizu.ac.jp/problems/ALDS1_12_A
最小全域木

僕が作成、提出したコードは、以下のとおりです。

Aizu Online Judge in C++ #ALDS1_12_A : Minimum Spanning Tree
/*
Aizu Online Judge in C++ #ALDS1_12_A : Minimum Spanning Tree
https://onlinejudge.u-aizu.ac.jp/problems/ALDS1_12_A
 提出コードの解答例
 https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
int main(void){
    int n,a[100][100],b[100]={},c[100]={};
    c[0]=1;
    cin>>n;
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            cin>>a[i][j];
        }
    }
    int k=1,m1,ans=0;
    while(1){
        int m=10000;
        for(int i=0;i<k;i++)
            for(int j=0;j<n;j++)
                if(a[b[i]][j]<m && a[b[i]][j]!=-1 && c[j]!=1){
                    m=a[b[i]][j];
                    m1=j;
                }
                if(k==n){
                    break;
                }
                b[k]=m1;
                c[m1]=1;
                ans+=m;
                k++;
    }
    cout<<ans<<endl;
    return 0;
}

設問の出典は、プログラミング問題のオンライン採点システム「Aizu Online Judge(AOJ)」です。
http://judge.u-aizu.ac.jp/onlinejudge/