C. 小花梨判连通(图的搜索)(map)

Description
小花梨给出𝑛个点,让𝑘位同学对这𝑛个点任意添加无向边,构成𝑘张图。小花梨想知道对于
每个点𝑖,存在多少个点𝑗(包括𝑖本身),使得𝑖和𝑗在这𝑘张图中都是连通的。
Input
第一行输入两个正整数𝑛和𝑘,分别表示点的个数和同学数。
接下来分成𝑘部分进行输入,每部分输入格式相同。
每部分第一行输入一个整数𝑎𝑖,表示第𝑖位同学连边的数目。
接下来𝑎𝑖行,每行两个正整数𝑢, 𝑣,表示第𝑖位同学将点𝑢和点𝑣之间进行连接。
可能会存在重边或者自环。
(1 ≤ 𝑛 ≤ 100000,1 ≤ 𝑘 ≤ 10,1 ≤ 𝑢, 𝑣 ≤ 𝑛, 0 ≤ 𝑎𝑖 ≤ 200000)
Output
输出𝑛行,第𝑖行输出在𝑘张图中都和编号为𝑖的点连通的点的数目(包括𝑖本身)
输入
4 2
3
1 2
1 3
2 3
2
1 2
3 4
输出
2
2
1
1

#include<iostream>
#include<vector>
#include<string.h>
#include<map>
using namespace std;
const int mod=1e9+7;
const int maxn=100009;
vector <int> c[maxn];
vector <int> G[100009];
map <vector<int>,int> mp;
int vis[100009];
long long ans[100009];
int n,k;

void dfs(int t,int now,int col){
	if(vis[now]==1)
	 return;
	vis[now]=1;
	c[now].push_back(col);
	for(int i=0;i<G[now].size();i++){
// if(vis[G[now][i]]==0)cout<<now<<"#"<<G[now][i]<<endl; 
		dfs(t,G[now][i],col);
	}
	return;
}

int main(){
	scanf("%d%d",&n,&k);
	for(int i=1;i<=k;i++){
		memset(G,0,sizeof(G));
		memset(vis,0,sizeof(vis));
		int num=1;
		int x,y;
		int n1;scanf("%d",&n1);
		for(int j=0;j<n1;j++){
		scanf("%d%d",&x,&y);
		G[x].push_back(y);
		G[y].push_back(x);
		}
		
		for(int j=1;j<=n;j++){
// cout<<vis[1]<<"@"<<endl;
			if(vis[j]==0){
				 dfs(i,j,num);
				 num++;
			}
			} 

// for(int i=1;i<=n;i++){
// cout<<"i: "<<i;
// for(int j=0;j<c[k][i].size();j++){
// cout<<c[k][i][j]<<"^";
// }
// cout<<"%"<<endl;
// } 
	}
// cout<<"*";
	for(int i=1;i<=n;i++){
		mp[c[i]]++;
	}
	
	for(int i=1;i<=n;i++)
	 printf("%lld\n",mp[c[i]]);		
} 

用dfs给每个点染色,这样就会得到一张n张图的颜色序列。用vector存这个序列,然后map一下就知道了。大佬们的方法果然强

bfs也行嘿嘿嘿

#include<iostream>
#include<vector>
#include<queue>
#include<string.h>
#include<map>
const int maxn=1e6+5;
using namespace std;
vector<int> G[maxn];
queue <int> q;
vector<int> c[maxn];
map<vector<int>,int> mp;
int vis[maxn];
int n,k;

void bfs(int now,int col){
	q.push(now);
	vis[now]=1;
	while(!q.empty()){
		int f=q.front();
// cout<<f<<"*"<<G[f].size()<<"* "<<col<<endl;
		q.pop();
		c[f].push_back(col);
		;
		for(int i=0;i<G[f].size();i++){
			if(vis[G[f][i]]==0){
				vis[G[f][i]]=1;
				q.push(G[f][i]);
			}
		}
	}
}

int main(){
	scanf("%d%d",&n,&k);
	for(int i=1;i<=k;i++){
		memset(G,0,sizeof(G));
		memset(vis,0,sizeof(vis));
		int t;
		scanf("%d",&t);
		for(int j=0;j<t;j++){
			int x,y;
			scanf("%d%d",&x,&y);
			G[x].push_back(y);
			G[y].push_back(x);
		}
		int cnt=1;
		for(int j=1;j<=n;j++){
			if(vis[j]==0){
			bfs(j,cnt);
			cnt++;
			}
		}
// for(int j=1;j<=n;j++){
// for(int b=0;b<c[j].size();b++) cout<<c[j][b];
// cout<<endl;
// }
	}
	for(int i=1;i<=n;i++){
		mp[c[i]]++;
	}
	for(int i=1;i<=n;i++) cout<<mp[c[i]]<<endl;
}
全部评论

相关推荐

点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务