UVA297 四分树 Quadtrees

四分树 Quadtrees

题面翻译

如图所示，可以用四分图来表示一个黑白图像，方法是用根节点表示整幅图像，然后把行列个分成两等份，按图中的方式编号，从左到右对应4个子节点。如果某子节点对应的区域全黑或全白，则直接用一个黑节点或白节点表示；如既有黑又有白，则用一个灰节点表示，并且为这个区域递归建树。
给出两棵四分树的先序遍历，求二者合并（黑色部分合并）黑像素的个数（每幅图都是32X32的）。p表示灰节点，f表示黑节点，e表示白节点。

题目描述

PDF

输入格式

输出格式

样例

样例输入

3
ppeeefpffeefe
pefepeefe
peeef
peefe
peeef
peepefefe

样例输出

There are 640 black pixels.
There are 512 black pixels.
There are 384 black pixels.

思路分析

根据图示易知，一整个黑块像素点为1024，题目定义的加法对应于两树的合并过程，分成几种情况讨论：
（1）两树的当前结点皆为黑色或白色则无需再往下递归合并，返回其中任意一个结点。
（2）若一黑一灰，则返回黑色结点；若一黑一白，同样返回黑色结点；若一灰一白，则返回灰色结点。
（3）若两个结点皆为灰色，则继续往下递归拆分成更小的子树，直到符合情况（1）（2）为止。

两树合并后得到新树，注意刚刚提及的信息，一整个黑块像素点为1024，每次进入其孩子都除以4，其对应当前结点黑块的像素点大小，遍历一次新树，累计求和即可。

代码实现

#include <iostream>
using namespace std;
struct Node{
    char val;
    Node *child[4];
    Node(char ch){
        val = ch;
        for(int i = 0; i < 4; i++){
            child[i] = nullptr;
        }
    }
};
Node* BuildTree(){
    char ch;
    cin >> ch;
    Node *root = new Node(ch);
    if(ch == 'p'){
        for(int i = 0; i < 4; i++){
            root->child[i] = BuildTree();
        }
    }
    return root;
}
Node* Merge(Node *b1, Node *b2){
    if(b1->val == 'e') return b2;
    else if(b1->val == 'f') return b1;
    else{
        if(b2->val == 'f') return b2;
        else if(b1->val == 'e') return b1;
        else{ //都是灰色结点
            for(int i = 0; i < 4; i++){
                b1->child[i] = Merge(b1->child[i], b2->child[i]);
            }
            return b1;
        }
    }
}
int CalPixels(Node *root, int pixels){
    if(root == nullptr) return 0;
    else if(root->val == 'f') return pixels;
    else if(root->val == 'p'){
        int ret = 0;
        for(int i = 0; i < 4; i++){
            ret += CalPixels(root->child[i], pixels / 4);
        }
        return ret;
    }else return 0;

}
int main(){
    int caseNumber;
    cin >> caseNumber;
    while(caseNumber--){
        Node *b1, *b2;
        b1 = BuildTree();
        b2 = BuildTree();
        b1 = Merge(b1, b2);
        int ans = CalPixels(b1, 1024); // 一整块像素是1024
        cout << "There are " << ans << " black pixels." << endl;
    }
    return 0;
}