二叉树的C++多输入

Question

我正在为如何处理二叉树的多个输入而苦苦挣扎。

输入格式为：

• 第一行采用整数n作为表示树级别的输入。
• 接下来的n+1行按级别顺序包含树中的节点。

输入示例：

2
1
2 3
4 5 6 7

以上输入的图形表示：

     1
    / \
   /   \
  2     3
 / \   / \
4   5 6   7

我想把从第二行到最后一行的值放入向量中，但是很难得到上面的输入。

下面是我尝试的：

int main(void) {
    vector<int> tree;
    int lv, num;
    int treeSize = 1;

    cin >> lv;

    for(int i = 0; i <= lv; i++) {
        while(tree.size() < treeSize && cin >> num) {
            tree.push_back(num);
        }
        treeSize += (i + 1) * 2;
    }

    return 0;
}

我的代码不能正常工作。例如，上面的示例输入必须有总共7个节点，但如果我在一行中放了太多的数字作为输入，它会超过7个节点。

在这种情况下，有没有适当的方法来获取输入？

Answer 1

代码一眼看上去是正确的。我会在调试器中跟踪它。你的代码有点难。我会像下面这样做数组填充。

int main(void) {
    vector<int> tree;
    int lv, num;

    cin >> lv;

    for(int i = 0, lvSize = 1; i <= lv; i++; lvSize *= 2) {
        for (int j = 0; j < lvSize; && cin >> num; j++) {
            tree.push_back(num);
        }
    }

    return 0;
}

因为你读了一个向量中的树，你就知道了元素的数量，代码就可以再简化和减少一点。

int main(void) {
    vector<int> tree;
    int lv, num;

    cin >> lv;
    int size = (1 << (lv + 1)) - 1;   // This is pow(2, lv + 1) - 1

    for(int i = 0; i < size; i++) {
        tree.push_back(num);
    }

    return 0;
}