MiYu原創(chuàng), 轉(zhuǎn)帖請(qǐng)注明 : 轉(zhuǎn)載自 ______________白白の屋
題目地址:
http://acm.hdu.edu.cn/showproblem.php?pid=1879
題目描述:
繼續(xù)暢通工程
Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 4134 Accepted Submission(s): 1602
Problem Description
省政府“暢通工程”的目標(biāo)是使全省任何兩個(gè)村莊間都可以實(shí)現(xiàn)公路交通（但不一定有直接的公路相連，只要能間接通過公路可達(dá)即可）?，F(xiàn)得到城鎮(zhèn)道路統(tǒng)計(jì)表，表中列出了任意兩城鎮(zhèn)間修建道路的費(fèi)用，以及該道路是否已經(jīng)修通的狀態(tài)。現(xiàn)請(qǐng)你編寫程序，計(jì)算出全省暢通需要的最低成本。
Input
測(cè)試輸入包含若干測(cè)試用例。每個(gè)測(cè)試用例的第1行給出村莊數(shù)目N ( 1< N < 100 )；隨后的 N(N-1)/2 行對(duì)應(yīng)村莊間道路的成本及修建狀態(tài)，每行給4個(gè)正整數(shù)，分別是兩個(gè)村莊的編號(hào)（從1編號(hào)到N），此兩村莊間道路的成本，以及修建狀態(tài)：1表示已建，0表示未建。
當(dāng)N為0時(shí)輸入結(jié)束。
Output
每個(gè)測(cè)試用例的輸出占一行，輸出全省暢通需要的最低成本。
Sample Input
3
1 2 1 0
1 3 2 0
2 3 4 0
3
1 2 1 0
1 3 2 0
2 3 4 1
3
1 2 1 0
1 3 2 1
2 3 4 1
0
Sample Output
3
1
0
題目分析:
這道題幾乎是和
HDU1233 <---詳情請(qǐng)點(diǎn)擊.
幾乎一模一樣, 唯一不同的是: 在一些村莊之間已經(jīng)有路存在了!!!!
這很好辦. 我們只需要在輸入的時(shí)候把已經(jīng)存在的路提前加入并查集就行了.
接下來的就 是 MST 了...........
代碼如下:
MiYu原創(chuàng), 轉(zhuǎn)帖請(qǐng)注明 : 轉(zhuǎn)載自 ______________白白の屋
#include <iostream>
#include <algorithm>
using namespace std;
typedef struct {
int parent;
int height;
}Tset;
typedef struct treeUFS{
public:
treeUFS(int n = 0):N(n+1) { set = new Tset[N];
visited = new bool[N];
for ( int i = 0; i != N; ++ i)
set[i].parent = i,set[i].height = 1,visited[i] = false;
}
~treeUFS(){ delete [] set; };
int find ( int x ){ int r = x; while ( r != set[r].parent ) r = set[r].parent;
return r;
}
void setVisit ( int x, int y ) { visited[x] = visited[y] = true; }
bool getVisit ( int x ) { return visited[x]; }
void Merge( int x,int y ){ x = find ( x ); y = find ( y );
if ( x == y ) return;
if ( set[x].height == set[y].height ){
set[y].parent = x;
set[x].height ++;
}
else if ( set[x].height < set[y].height ) {
set[x].parent = y;
}
else{ set[y].parent = x;
}
}
private:
Tset *set;
bool *visited;
int N;
}treeUFS;
typedef struct edge {
int v1,v2;
int wei;
int isBuild;
}EDGE;
EDGE edge[5005];
bool cmp ( EDGE A, EDGE B )
{
return A.wei < B.wei;
}
int main ()
{
int N;
while ( scanf ( "%d",&N ) , N )
{
int n = N * ( N - 1 ) / 2;
int v1,v2,b;
memset ( edge, 0 , sizeof ( edge ) );
treeUFS UFS ( N );
for ( int i = 1; i <= n; ++ i )
{
scanf ( "%d%d", &edge[i].v1,&edge[i].v2 );
scanf ( "%d%d", &edge[i].wei,&edge[i].isBuild );
if ( edge[i].isBuild == 1 )
{
UFS.Merge ( edge[i].v1, edge[i].v2 );
UFS.setVisit ( edge[i].v1, edge[i].v2 );
}
}
sort ( edge + 1, edge + n + 1, cmp );
int sum = 0;
for ( int i = 1; i <= n; ++ i )
{
if ( ( !UFS.getVisit(edge[i].v1) || !UFS.getVisit(edge[i].v2) ) || UFS.find(edge[i].v1) != UFS.find(edge[i].v2) )
{
UFS.setVisit ( edge[i].v1, edge[i].v2 );
UFS.Merge ( edge[i].v1, edge[i].v2 );
sum += edge[i].wei;
}
}
printf ( "%d\n",sum );
}
return 0;
}