二叉树之中序遍历的递归和迭代实现

  1. 迭代 --- O(n) time and O(n) space;
  2. 递归 --- O(n) time and O(n) space ;
  3. Morris traversal --- O(n) time and O(1) space!!!

迭代算法:

vector inorderTraversal(TreeNode* root) {
	vector res;
	std::stack stn;
	while (root || !stn.empty()) {
		while (root) {
			stn.push(root);
			root = root->left;
		}
		root = stn.top(); stn.pop();
		res.push_back(root->val);
		root = root->right;
	}
	return res;
}




递归实现:

vector res;
vector inorderTraversal(TreeNode* root) {
	if (root) {
		inorderTraversal(root->left);
		res.push_back(root->val);
		inorderTraversal(root->right);
	}
	return res;
}
Morris traversal:

vector inorderTraversal(TreeNode* root) {
    TreeNode* curNode = root;
    vector nodes;
    while (curNode) {
        if (curNode -> left) {
            TreeNode* predecessor = curNode -> left;
            while (predecessor -> right && predecessor -> right != curNode)
                predecessor = predecessor -> right;
            if (!(predecessor -> right)) {
                predecessor -> right = curNode;
                curNode = curNode -> left;
            }
            else {
                predecessor -> right = NULL;
                nodes.push_back(curNode -> val);
                curNode = curNode -> right;
            }
        }
        else {
            nodes.push_back(curNode -> val);
            curNode = curNode -> right;
        }
    }
    return nodes;
}



你可能感兴趣的:(c/c++)