PAT 天梯杯 L2-024 部落 dfs,连通块
L2-024. 部落
时间限制
120 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
陈越
在一个社区里,每个人都有自己的小圈子,还可能同时属于很多不同的朋友圈。我们认为朋友的朋友都算在一个部落里,于是要请你统计一下,在一个给定社区中,到底有多少个互不相交的部落?并且检查任意两个人是否属于同一个部落。
输入格式:
输入在第一行给出一个正整数N(<= 104),是已知小圈子的个数。随后N行,每行按下列格式给出一个小圈子里的人:
K P[1] P[2] ... P[K]
其中K是小圈子里的人数,P[i](i=1, .., K)是小圈子里每个人的编号。这里所有人的编号从1开始连续编号,最大编号不会超过104。
之后一行给出一个非负整数Q(<= 104),是查询次数。随后Q行,每行给出一对被查询的人的编号。
输出格式:
首先在一行中输出这个社区的总人数、以及互不相交的部落的个数。随后对每一次查询,如果他们属于同一个部落,则在一行中输出“Y”,否则输出“N”。
输入样例:4 3 10 1 2 2 3 4 4 1 5 7 8 3 9 6 4 2 10 5 3 7输出样例:
10 2 Y N
题目是求连通块及两点是否在一个连通块中,因为题目给的空间有点小所以要用vector不然会内存超限
#include <algorithm> #include <iostream> #include <cstring> #include <cstdio> #include <vector> #include <cmath> #include <map> #define maxn 10010 #define debug(a) cout << #a << " " << a << endl using namespace std; typedef long long ll; vector<int> E[maxn]; int a[maxn], vis[maxn], m, cnt; int dx[] = { 0, 0, 1, -1 }, dy[] = { 1, -1, 0, 0 }; void init() { for( int i = 0; i < maxn; i ++ ) { E[i].clear(); } for( int i = 0; i < maxn; i ++ ) { vis[i] = 0; } m = -1; } void dfs( int x ) { for( int i = 0; i < E[x].size(); i ++ ) { if( !vis[E[x][i]] ) { /*debug(i); debug(E[x][i]);*/ vis[E[x][i]] = cnt; dfs(E[x][i]); } } } int main() { int T; cin >> T; init(); while( T -- ) { int k; cin >> k; for( int i = 0; i < k; i ++ ) { cin >> a[i]; if( a[i] > m ) { m = a[i]; } for( int j = 0; j < i; j ++ ) { E[a[i]].push_back(a[j]); E[a[j]].push_back(a[i]); } } } cin >> T; cnt = 1; for( int i = 1; i <= m; i ++ ) { if( !vis[i] ) { vis[i] = cnt; dfs(i); //将在同一个连通块中的点的vis值变成同一个值,这样有多少个不同的值就有多少个不同的连通块,用cnt直接记下来 //cout << "---" << i << endl; cnt ++; } } cout << m << " " << cnt - 1 << endl; while( T -- ) { int x, y; cin >> x >> y; if( vis[x] == vis[y] ) { cout << "Y" << endl; } else { cout << "N" << endl; } } return 0; }
彼时当年少,莫负好时光。