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;
}