Gym - 101669D Harry Potter and The Vector Spell (最小生成树)

Gym - 101669D Harry Potter and The Vector Spell (最小生成树)_第1张图片

 

题意:求一个01矩阵的秩,这个矩阵的每一列只有两个1.

解题思路:很容易联想到图论,我们对于每一列的两个一所在的行连边,最后形成一个图,最后的秩肯定是这个图最大的一个无环子图,其实就是最小生成树。

 

#include 
#include 
#include 
#include 
using namespace std;
const int maxn=1e5+5;
int fa[maxn];
vectorP[maxn];
int get(int w){
	return fa[w]==w?w:fa[w]=get(fa[w]);
}
int main(){
	int n,m;
	int ans=0,x,k;
	scanf("%d%d",&m,&n);
	for (int i=1;i<=m;i++){
		scanf("%d",&k);
		while (k--){
			scanf("%d",&x);
			P[x].push_back(i);
		}
	}
	for (int i=1;i<=m;i++) fa[i]=i;
	for (int i=1;i<=n;i++){
		int u=P[i][0],v=P[i][1];
		int fx=get(u);
		int fy=get(v); 
		if (fy!=fx){
			ans++;
			fa[fx]=fy;
		}
	}
	printf("%d\n",ans);
}

 

 

 

你可能感兴趣的:(————ACM相关————,——图论相关——,ACM,-,最小生成树)