1365：FBI樹(fbi)

【題目描述】

我們可以把由“00”和“11”組成的字符串分爲三類：全“00”串稱爲BB串，全“11”串稱爲II串，既含“00”又含“11”的串則稱爲FF串。

FBI樹是一種二叉樹，它的結點類型也包括FF結點，BB結點和II結點三種。由一個長度爲2N2N的“0101”串SS可以構造出一棵FBI樹TT，遞歸的構造方法如下：

TT的根結點爲RR，其類型與串SS的類型相同；

若串SS的長度大於11，將串SS從中間分開，分爲等長的左右子串S1S1和S2S2；由左子串S1S1構造RR的左子樹T1T1，由右子串S2S2構造RR的右子樹T2T2。

現在給定一個長度爲2N2N的“0101”串，請用上述構造方法構造出一棵FBI樹，並輸出它的後序遍歷序列。

【輸入】

第一行是一個整數N（0≤N≤10）N（0≤N≤10），第二行是一個長度爲2N2N的“0101”串。

【輸出】

一行，這一行只包含一個字符串，即FBI樹的後序遍歷序列。

【輸入樣例】

3
10001011

【輸出樣例】

IBFBBBFIBFIIIFF

【提示】

對於40%的數據，N≤2N≤2；

對於100%的數據，N≤10N≤10。

 

#include <bits/stdc++.h>
using namespace std;

struct Node {
    char value;
    Node *left, *right;
};

char GetValue(const string &s, int index, int limit)
{
    int zero = 0, one = 0;
    for (int i = index; i < limit; i++) {
        if (s[i] == '0') {
            zero += 1;
        } else if (s[i] == '1') {
            one += 1;
        }
    }
    if (zero > 0 && one > 0) {
        return 'F'; // 01
    } else if (zero > 0) {
        return 'B'; // 0
    } else if (one > 0) {
        return 'I'; // 1
    } else {
        return 'E'; // error
    }
}

Node *CreateTree(const string &s, int left, int right)
{
    if (left < 0 || left >= right) {
        return NULL;
    } else {
        Node *root = new Node;
        root->value = GetValue(s, left, right);
        // cout << left << "," << right << "=" << root->value << endl;
        root->left = root->right = NULL;
        if (right - left > 1) { // 多於1個
            int mid = (left + right) / 2;
            root->left = CreateTree(s, left, mid);
            root->right = CreateTree(s, mid, right);
        }
        return root;
    }
}

string PreOrder(Node *root)
{
    string s;
    if (root != NULL) {
        s.push_back(root->value);
        s += PreOrder(root->left);
        s += PreOrder(root->right);
    }
    return s;
}

string InOrder(Node *root)
{
    string s;
    if (root != NULL) {
        s += InOrder(root->left);
        s.push_back(root->value);
        s += InOrder(root->right);
    }
    return s;
}

string PostOrder(Node *root)
{
    string s;
    if (root != NULL) {
        s += PostOrder(root->left);
        s += PostOrder(root->right);
        s.push_back(root->value);
    }
    return s;
}

int main()
{
    // freopen("1.txt", "r", stdin);
    int n = 3;
    string s = "10001011";
    cin >> n >> s;
    Node *root = CreateTree(s, 0, s.size());
    // cout << PreOrder(root) << endl;
    // cout << InOrder(root) << endl;
    cout << PostOrder(root) << endl;
    return 0;
}