A. 修路
时间限制 3000 ms 内存限制 65536
KB
题目描述
小弱的学校很喜欢修路,现在给你一张他学校的地图,地图上有n个点和m条双向边,每条边代表一条路,这条路有可能是畅通,也有可能正在修路。大家都知道修路使得交通很不方便。所有小弱很想学校快快的把路修好,使得他能够很轻松的到达主楼915去刷题。但考虑到学校的施工能力有限,小弱想让你帮他算出学校需要集中力量马上修好的最少路数,使得他能够从学校任意点出发,在不经过正在施工的路下到达主楼(编号为1)。
输入格式
有多组数据。
每组数据以n( 1<=n<=10000), m(1<=m<=200000)开头。接下来一行有m行数。每行有三个数,对应于u, v, s,分别为这条路的两端点(编号从1到n)和路况,s = 0代表畅通, s = 1 代表正在修路。输入保证图是连通图。
输出格式
对每组数据输出对应的最少路数。
输入样例
3 2
1 2 0
1 3 1
3 2
1 2 0
1 3 0
3 2
1 2 1
1 3 1
输出样例
1
0
2
题目说明输入保证图是联通图。于是只需要考虑修路与否,如果不修路,对该点跑并查集即可;在修路则不管。然后检查所有点,依次检查是否入团。
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<map>
#include<cstdlib>
using namespace std;
#define M 100500
int p[M],vi[M];
int n, m;
int u,v,s,ans;
int find(int x)
{
if(x!=p[x])
{
p[x]=find(p[x]);
}
return p[x];
}
void init()
{
for(int i=1; i<=n; i++)
{ p[i]=i;}
while(m--)
{
scanf("%d%d%d",&u,&v,&s);
if(s==0)
p[find(v)] = find(u);
}
ans = 0;
memset(vi,-1, sizeof(vi));
}
void solve()
{
for(int i=1; i<=n; i++)
if(vi[find(i)]==-1)
{
ans++;
vi[find(i)]=1;
}
printf("%d\n",ans-1);
}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
init();
solve();
}
}