HDU-1879は引き続き工事を円滑化する(最小生成ツリー[prim])

2571 ワード

工事を続ける
Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Problem Description
省政府の「円滑な工事」の目標は、全省のどの2つの村の間でも道路交通を実現することである.(ただし、直接の道路がつながっているとは限らない.間接的に道路を通っていればよい).現在、都市道路統計表を入手し、任意の2つの都市間で道路を建設する費用と、その道路がすでに開通しているかどうかの状態をリストした.プログラムを作成して、全省の円滑化に必要な最低コストを計算してください.

Input
テスト入力には、いくつかのテスト例が含まれます.各試験例の1行目は、村の数N(1Nが0のとき入力は終了します.

Output
各テスト例の出力は1行を占め、全省の円滑化に必要な最低コストを出力する.

Sample Input

Sample Output

最初は、作成したエッジの点を直接キューに押し込もうとしたが、面倒な感じがした.次に、エッジがすでに存在する場合、その重み値を0に設定し、裸の最小生成ツリーになることを考えます.

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

using namespace std;

const int INF=0x3f3f3f3f;

struct Edge {
    int v,w;
    Edge(int vv=0,int ww=INF):v(vv),w(ww) {}
    bool operator < (const Edge& a) const {
        return w>a.w;
    }
}u;

int n,cnt,ans,vis[105],dis[105];
priority_queue <Edge> q;
vector<vector<Edge> > g;

void Prim() {
    int i,j,v,w;
    q.push(Edge(1,0));
    memset(vis,false,sizeof(vis));
    memset(dis,0x3f,sizeof(dis));
    ans=cnt=0;
    while(!q.empty()) {
        u=q.top();
        q.pop();
        if(!vis[u.v]) {
            vis[u.v]=true;
            ++cnt;
            ans+=u.w;
            for(i=0,j=g[u.v].size();i<j;++i)
                if(!vis[v=g[u.v][i].v]&&dis[v]>(w=g[u.v][i].w)) {
                    dis[v]=w;
                    q.push(Edge(v,w));
                }
        }
    }
}

int main() {
    int i,s,e,cst,cond,t;
    while(scanf("%d",&n),n) {
        g.clear();
        g.resize(n+1);
        for(i=0,t=(n*(n-1))>>1;i<t;++i) {
            scanf("%d%d%d%d",&s,&e,&cst,&cond);
            g[s].push_back(Edge(e,cond==1?0:cst));//      ，    0
            g[e].push_back(Edge(s,cond==1?0:cst));
        }
        Prim();
            printf("%d
",ans);
    }
    return 0;
}

linux同期ツール

[TreasureData]tdコマンド実行でTreasureData::AuthErrorエラーになる場合の対処