1091. Acute Stroke (30)解题报告

最新推荐文章于 2021-08-28 19:53:43 发布

信道者

最新推荐文章于 2021-08-28 19:53:43 发布

阅读量307

点赞数

CC 4.0 BY-SA版权

分类专栏： PAT 文章标签： C++ PAT

本文链接：https://blog.youkuaiyun.com/chr1991/article/details/53909085

PAT 专栏收录该内容

173 篇文章

订阅专栏

本文介绍了一种在三维空间中使用并查集算法解决特定问题的方法。通过三维矩阵表示空间，并利用并查集进行元素合并及计数，解决了求不小于指定大小的集合内元素数量的问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这实际上是并查集问题。题目要求的输出就是不小于T大小的集合里面含有的所有元素的数量。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
using namespace std;

void setunion(int s[], int i, int j);
int findroot(int s[], int i);

int main(int argc, char** argv) {
	int *s, m, n, l, t, i, j, k, tmp, index, cnt;
	bool *matrix;
	scanf("%d %d %d %d", &m, &n, &l, &t);
	s = new int[m * n * l];
	matrix = new bool[m * n * l];

	for(i = 0; i < l; i++){
		for(j = 0; j < m; j++){
			for(k = 0; k < n; k++){
				s[i * m * n + j * n + k] = -1;
			}
		}
	}
	
	for(i = 0; i < l; i++){
		for(j = 0; j < m; j++){
			for(k = 0; k < n; k++){
				scanf("%d", &tmp);
				if(tmp){
					matrix[i * m * n + j * n + k] = true;
				}
				else{
					matrix[i * m * n + j * n + k] = false;
				}
			}
		}
	}
	
	for(i = 0; i < l; i++){
		for(j = 0; j < m; j++){
			for(k = 0; k < n; k++){
				index = i * m * n + j * n + k;
				if(matrix[index]){
					if(i - 1 >= 0 && matrix[(i - 1) * m * n + j * n + k]){
						setunion(s, index, (i - 1) * m * n + j * n + k);
					}
					if(j - 1 >= 0 && matrix[i * m * n + (j - 1) * n + k]){
						setunion(s, index, i * m * n + (j - 1) * n + k);
					}
					if(k - 1 >= 0 && matrix[i * m * n + j * n + k - 1]){
						setunion(s, index, i * m * n + j * n + k - 1);
					}
				}
			}
		}
	}
	cnt = 0;
	for(i = 0; i < l; i++){
		for(j = 0; j < m; j++){
			for(k = 0; k < n; k++){
				index = i * m * n + j * n + k;
				if(matrix[index] && s[index] < 0 && -s[index] >= t){
					cnt += (-s[index]);
				}	
			}
		}

	}
	
	printf("%d", cnt);
	
	delete[] s;
	delete[] matrix;
	return 0;
}

void setunion(int s[], int i, int j){
	int root1, root2;
	root1 = findroot(s, i);
	root2 = findroot(s, j);
	if(root1 == root2){
		return;
	}
	if(s[root1] < s[root2]){
		s[root1] += s[root2];
		s[root2] = root1;
	}
	else{
		s[root2] += s[root1];
		s[root1] = root2;
	}
	return;
}
int findroot(int s[], int i){
	if(s[i] < 0){
		return i;
	}
	else{
		return s[i] = findroot(s, s[i]);
	}
}