图论模板 连通分量数 DFS or 并查集
C Battle Over Cities (25 分)(DFS)
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int num=2000005;//注意:链式前向星,无向图,开两倍(最大边数1000*1000的两倍)!!!
int n,m,k,a,b,c,ans=0;//ans为连通分量数量
struct edge
{
int to,next;
}edge[num];
int cnt,head[num];
void init()
{
for(int i=0;i<num;i++)
{
edge[i].next=-1;
head[i]=-1;
}
cnt=0;
}
void addedge(int u,int v)
{
edge[cnt].to=v;
edge[cnt].next=head[u];
head[u]=cnt++;
}//链式前向星存图
int vis[1010],flag;
void dfs(int x)
{
for(int i=head[x];i!=-1;i=edge[i].next)
if(edge[i].to!=c)
{
if(vis[edge[i].to]==0)
{
vis[edge[i].to]=1;
dfs(edge[i].to);
}
}
}
int main()
{
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
cin>>n>>m>>k;
init();
for(int i=1;i<=m;i++)
{
cin>>a>>b;
addedge(a,b);
addedge(b,a);
}
for(int i=1;i<=k;i++)
{
memset(vis,0,sizeof(vis));
ans=0;
cin>>c;
for(int i=1;i<=n;i++)
{
if(vis[i]==0&&i!=c)
{
vis[i]=1;
ans++;
dfs(i);
}
}
cout<<ans-1<<endl;
}
return 0;
}
C Battle Over Cities (25 分)(并查集)
#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
const int maxn=1010;
vector<int> G[maxn];//邻接表
int n,m,k,a,b,c,p[maxn],vis[maxn];
int find(int x)
{
return p[x]==x?x:p[x]=find(p[x]);
}//并查集
void unio(int a,int b)
{
int x=find(a),y=find(b);
if(x!=y) p[x]=y;
}
void init()
{
for(int i=1;i<=n;i++)
{
p[i]=i;
vis[i]=0;
}
}
int main()
{
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
cin>>n>>m>>k;
for(int i=0;i<m;i++)
{
cin>>a>>b;
G[a].push_back(b);
G[b].push_back(a);
}
for(int t=0;t<k;t++)
{
cin>>c;
init();
for(int i=1;i<=n;i++)
{
int size=G[i].size();
for(int j=0;j<size;j++)
{
if(i!=c&&G[i][j]!=c)
unio(i,G[i][j]);
}
}
int ans=0;
/*
for(int i=1;i<=n;i++)
if(i!=c) find(i);//注意是find(i)不是p[i]
sort(p+1,p+n+1);
for(int i=1;i<=n;i++)
if(p[i]!=p[i-1]&&p[i]!=c)
ans++;
cout<<ans-1<<endl;
*/
for(int i=1;i<=n;i++)
{
if(i==c) continue;
int fa=find(i);
if(!vis[fa])
{
ans++;
vis[fa]=1;
}
}
cout<<ans-1<<endl;
}
return 0;
}
F Interstellar Love(DFS)
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int num=20005;
struct edge
{
int to,next;
}edge[num];
int cnt,head[num];
void init()
{
for(int i=0;i<num;i++)
{
edge[i].next=-1;
head[i]=-1;
}
cnt=0;
}
void addedge(int u,int v)
{
edge[cnt].to=v;
edge[cnt].next=head[u];
head[u]=cnt++;
}
int vis[10005],flag;
void dfs(int x,int fa)
{
for(int i=head[x];i!=-1;i=edge[i].next)
if(edge[i].to!=fa)
{
if(vis[edge[i].to]==0)
{
vis[edge[i].to]=1;
dfs(edge[i].to,x);
}
else
flag=0;
}
}
int main()
{
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
int T;
cin>>T;
for(int t=1;t<=T;t++)
{
int n,m,a,b,ans=0,fix=0;
cin>>n>>m;
init();
memset(vis,0,sizeof(vis));
for(int i=1;i<=m;i++)
{
cin>>a>>b;
addedge(a,b);
addedge(b,a);
}
for(int i=1;i<=n;i++)
if(vis[i]==0)
{
vis[i]=1;
flag=1;
if(head[i]!=-1)
{
ans++;
dfs(i,0);
if(flag==0)
fix++;
}
}
printf("Night sky #%d: %d constellations, of which %d need to be fixed.\n\n",t,ans,fix);
}
return 0;
}
F Interstellar Love(并查集)
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int num=20005;
int vis[num],fanode[num],faedge[num];//点的祖先节点,边的祖先节点
int havenode[num],haveedge[num];//祖先节点连接点数,祖先节点连接边数
int find(int x)
{
return x==fanode[x]?x:fanode[x]=find(fanode[x]);
}
int main()
{
int T;
cin>>T;
for(int t=1;t<=T;t++)
{
int n,m,a,b,ans=0,fix=0;
cin>>n>>m;
for(int i=1;i<=n;i++)
{
fanode[i]=i;
vis[i]=havenode[i]=haveedge[i]=0;
}
for(int i=1;i<=m;i++)
{
cin>>a>>b;
vis[a]=vis[b]=1;
int x=find(a),y=find(b);
if(x!=y)
fanode[x]=fanode[y];
faedge[i]=fanode[y];
}
for(int i=1;i<=n;i++)
havenode[find(fanode[i])]++;
for(int i=1;i<=m;i++)
haveedge[find(faedge[i])]++;
//for(int i=1;i<=n;i++)
// cout<<havenode[i]<<" "<<haveedge[i]<<endl;
for(int i=1;i<=n;i++)
{
if(vis[i]&&havenode[i])
{
ans++;//至少有两个结点的连通分量
if(havenode[i]<=haveedge[i])
fix++;//存在环的连通分量
}
}
printf("Night sky #%d: %d constellations, of which %d need to be fixed.\n\n",t,ans,fix);
}
return 0;
}