26、二叉搜索树与双向链表 好题,值得再看一遍
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
0、最笨的一种写法,这也是最容易理解的一种方法了
中序遍历二叉树,然后用一个数组类保存遍历的结果,这样在数组中节点就按顺序保存了,然后再来修改指针,虽然没有一点技术含量,但是最后竟然还通过了 哈哈哈。。。
TreeNode* Convert(TreeNode* pRootOfTree) { if (pRootOfTree == NULL) return pRootOfTree; vector<TreeNode*> result; Convert(pRootOfTree, result); return Convert(result); } void Convert(TreeNode* node,vector<TreeNode*>&result) { if (node->left != nullptr) Convert(node->left, result); result.push_back(node); if (node->right != nullptr) Convert(node->right, result); } TreeNode* Convert(vector<TreeNode*>& result) { for (int i = 0; i < result.size()-1; ++i) { result[i]->right = result[i + 1]; result[i+1]->left = result[i]; } return result[0]; }
0-1借助栈和数组类进行数据保存,最后修改指针指向
关键在于二叉树的层次遍历这一块
public: TreeNode* Convert(TreeNode* pRootOfTree) { if (pRootOfTree == nullptr) return nullptr; vector<TreeNode*> result; stack<TreeNode*> s; // 形成一个中序遍历的结果,并添加指针。 while (!s.empty() || pRootOfTree != nullptr) { if (pRootOfTree != nullptr) { s.push(pRootOfTree); pRootOfTree = pRootOfTree->left; } else { pRootOfTree = s.top(); s.pop(); result.push_back(pRootOfTree); pRootOfTree = pRootOfTree->right; } } // 修改链表指针指向。 for (int i = 0; i < result.size() - 1; ++i) { result[i + 1]->left = result[i]; result[i]->right = result[i+1]; } return result[0]; }
1、借助栈进行节点保存,很厉害的一种写法
我服啦,采用的是跟剑指offer上一样的写法
- 明确Convert函数的功能。
输入:输入一个二叉搜索树的根节点。
过程:将其转化为一个有序的双向链表。
输出:返回该链表的头节点。 - 明确成员变量pLast的功能。
pLast用于记录当前链表的末尾节点。 - 明确递归过程。
递归的过程就相当于按照中序遍历,将整个树分解成了无数的小树,然后将他们分别转化成了一小段一小段的双向链表。再利用pLast记录总的链表的末尾,然后将这些小段链表一个接一个地加到末尾。
TreeNode* Convert(TreeNode* pRootOfTree) { TreeNode* head = NULL, * pre = NULL;//head 输出,pre记录上一次出栈值 stack<TreeNode*> s; while (pRootOfTree || !s.empty()) { while (pRootOfTree!=nullptr) { s.push(pRootOfTree); pRootOfTree = pRootOfTree->left; } if (!s.empty()) { pRootOfTree = s.top(); s.pop(); if (pre != NULL) { pre->right = pRootOfTree; pRootOfTree->left = pre; } else//pre为空,表示s第一次出栈,第一次出栈值为最左值,即输出值 { head = pRootOfTree; } pre = pRootOfTree;