Я пишу рекурсивный обход предварительного заказа для следующего двоичного дерева: [введите описание изображения здесь] [1].
Двоичное дерево установлено как line = "[1,2,3,4,5,6,7,null,8,9,10,11,null,12,13]";
в int main()
.
Я написал свое решение в функции:
class Solution {
private:
void traverse(TreeNode* root, vector<int> answer){
if (root == NULL){
return;
}
answer.push_back(root->val);
traverse(root->left,answer);
traverse(root->right,answer);
}
public:
vector<int> preorderTraversal(TreeNode* root){ //function preorderTraversal() output type is vector<int>
vector<int> answer;
answer.clear();
traverse(root,answer);
return answer;
}
};
Переменная root
- это двоичное дерево, заданное переменной line
.
Переменная answer
- это возвращаемое значение обхода предварительного заказа.
Оба они являются входными аргументами рекурсивной функции void traverse(TreeNode* root, vector<int> answer)
.
Рекурсивная функция traverse
установлена как закрытая функция-член.
Я ожидаю, что результат будет:
[1, 2, 4, 8, 5, 9, 10, 3, 6, 11, 7, 12, 13]
Program ended with exit code: 0
Однако, когда я запустил этот код с Xcode, я получил следующий вывод:
[]
Program ended with exit code: 0
Я обнаружил, что значения в answer
будут удалены, когда мои обходные узлы вернутся к своим корням!
Я могу заставить его работать, только удалив переменную answer
в качестве аргумента для рекурсивной функции и установив ее как переменная publi c как это:
class Solution {
private:
void traverse(TreeNode* root){
if (root == NULL){
return;
}
answer.push_back(root->val);
traverse(root->left);
traverse(root->right);
}
public:
vector<int> answer;
vector<int> preorderTraversal(TreeNode* root){ //function preorderTraversal() output type is vector<int>
answer.clear();
traverse(root);
return answer;
}
};
Мой вопрос: почему рекурсивный Функция traverse
не будет сохранять результаты , присвоенные answer
, когда я использую эту переменную в качестве аргумента?
Спасибо!
Полный код такой следует:
#include <iostream>
#include <sstream>
#include <vector>
#include <stack>
#include <queue>
#include <string>
using namespace std;
struct TreeNode
{
// member vars
int val;
TreeNode* left;
TreeNode* right;
// constructor
TreeNode(int x): val(x), left(NULL), right(NULL) {} //Initialize val=x, left=NULL, right=NULL, x is the value of root
};
class Solution {
private:
void traverse(TreeNode* root, vector<int> answer){
if (root == NULL){
return;
}
answer.push_back(root->val);
traverse(root->left,answer);
traverse(root->right,answer);
}
public:
//vector<int> answer;
vector<int> preorderTraversal(TreeNode* root){ //function preorderTraversal() output type is vector<int>
vector<int> answer;
answer.clear();
traverse(root,answer);
return answer;
}
};
void trimLeftTrailingSpaces(string &input) {
input.erase(input.begin(), find_if(input.begin(), input.end(), [](int ch) {
return !isspace(ch);
}));
}
void trimRightTrailingSpaces(string &input) {
input.erase(find_if(input.rbegin(), input.rend(), [](int ch) {
return !isspace(ch);
}).base(), input.end());
}
TreeNode* stringToTreeNode(string input) {
trimLeftTrailingSpaces(input);
trimRightTrailingSpaces(input);
input = input.substr(1, input.length() - 2);
if (!input.size()) {
return nullptr;
}
string item;
stringstream ss;
ss.str(input);
getline(ss, item, ',');
TreeNode* root = new TreeNode(stoi(item));
queue<TreeNode*> nodeQueue;
nodeQueue.push(root);
while (true) {
TreeNode* node = nodeQueue.front();
nodeQueue.pop();
if (!getline(ss, item, ',')) {
break;
}
trimLeftTrailingSpaces(item);
if (item != "null") {
int leftNumber = stoi(item);
node->left = new TreeNode(leftNumber);
nodeQueue.push(node->left);
}
if (!getline(ss, item, ',')) {
break;
}
trimLeftTrailingSpaces(item);
if (item != "null") {
int rightNumber = stoi(item);
node->right = new TreeNode(rightNumber);
nodeQueue.push(node->right);
}
}
return root;
}
string integerVectorToString(vector<int> list, int length = -1) {
if (length == -1) {
length = list.size();
}
if (length == 0) {
return "[]";
}
string result;
for(int index = 0; index < length; index++) {
int number = list[index];
result += to_string(number) + ", ";
}
return "[" + result.substr(0, result.length() - 2) + "]";
}
int main() {
string line;
//while (getline(cin, line)) {
line = "[1,2,3,4,5,6,7,null,8,9,10,11,null,12,13]";
TreeNode* root = stringToTreeNode(line);
vector<int> ret = Solution().preorderTraversal(root);
string out = integerVectorToString(ret);
cout << out << endl;
//}
return 0;
}