UVa 297 - Quadtrees

题目:利用四叉树处理图片,给你两张黑白图片的四叉树,问两张图片叠加后黑色的面积。

分析:搜索、数据结构。把图片分成1024块1*1的小正方形,建立一位数组记录对应小正方形的颜色。

            利用递归根据字符串,建立相应四叉树。在建树的过程中,树节点计算当前节点对应的小正方形

            编号区间。这里处理类似于线段树,将父节点的区间等分成4份分别对应四棵子树的编号区间。

            建树到达叶子时(color为‘f’或者‘e’),直接将颜色数组赋值即可。当树建完时,颜色数组即染色

            完毕。将两棵树依次染色到同一数组,统计黑色节点个数即可。

注意:数组大小,防止RE。

#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <cmath>

using namespace std;

int Color[1024];

typedef struct tnode
{
	char   color;
	tnode* child[4];
}tnode;
tnode *root;
tnode  Node[1365];
int    Count;

tnode* dfs( char* word, int &move, int a, int b )
{
	tnode* root = &Node[Count ++];
	root->color = word[move ++];
	if ( root->color == 'p' ) {
		int mid = (a+b)>>1;
		int m_l = (a+mid)>>1;
		int m_r = (mid+b)>>1;
		//构建四个子树 
		root->child[0] = dfs( word, move, a, m_l );
		root->child[1] = dfs( word, move, m_l+1, mid );
		root->child[2] = dfs( word, move, mid+1, m_r );
		root->child[3] = dfs( word, move, m_r+1, b );
	}else if ( root->color == 'f' ) {
		//叶子节点染色 
		for ( int i = a ; i <= b ; ++ i )
			Color[i] = 1;
	}return root;
}

int main()
{
	int  n,move;
	char data[1366];
	
	while ( ~scanf("%d",&n) )
	for ( int i = 0 ; i < n ; ++ i ) {
		for ( int i = 0 ; i < 1024 ; ++ i )
			Color[i] = 0;
		
		//处理图片1	
		scanf("%s",data);
		Count = 0;
		dfs( data, move = 0, 0, 1023 );
		
		//处理图片2 
		scanf("%s",data);
		Count = 0;
		dfs( data, move = 0, 0, 1023 );
		
		//统计黑色节点 
		int sum = 0;
		for ( int i = 0 ; i < 1024 ; ++ i )
			sum += Color[i];
		printf("There are %d black pixels.\n",sum);
	}
	
	return 0;
}

            

你可能感兴趣的:(UVa 297 - Quadtrees)