UVa 297 - Quadtrees

2014-11-23 23:55:27 · 作者: · 浏览: 2
题目:利用四叉树处理图片,给你两张黑白图片的四叉树,问两张图片叠加后黑色的面积。
分析:搜索、数据结构。把图片分成1024块1*1的小正方形,建立一位数组记录对应小正方形的颜色。
利用递归根据字符串,建立相应四叉树。在建树的过程中,树节点计算当前节点对应的小正方形
编号区间。这里处理类似于线段树,将父节点的区间等分成4份分别对应四棵子树的编号区间。
建树到达叶子时(color为‘f’或者‘e’),直接将颜色数组赋值即可。当树建完时,颜色数组即染色
完毕。将两棵树依次染色到同一数组,统计黑色节点个数即可。
注意:数组大小,防止RE。
#include   
#include   
#include   
#include   
#include   
  
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; }