题目:利用四叉树处理图片,给你两张黑白图片的四叉树,问两张图片叠加后黑色的面积。
分析:搜索、数据结构。把图片分成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;
}