自己今天真是笨死了,这么简单的题目,竟然在当场写代码的时候没有写出来。
这里写一下,其实这个题目非常简单,就是一个中序遍历,在中序遍历的时候,更改一下指针指向。自己在面试写的时候太紧张了,就蒙了。。。
下面贴一下代码实现:
TreeNode* inorderFlatTree(TreeNode * root){
stack<TreeNode *> st;
TreeNode * temp = root;
TreeNode * sw = NULL;
while (temp != NULL || !st.empty()) {
while (temp != NULL) {
st.push(temp);
temp = temp->left;
}
if (!st.empty()) {
if (sw != NULL) {
sw->right = st.top();
st.top()->left = sw;
}
cout<<st.top()->val<<endl;
temp = st.top()->right;
sw = st.top();
st.pop();
}
}
return sw;
}
这样子返回的是链表的尾部,如果需要增序的话,用个while循环,把sw指向头结点就可以了。
while(sw != NULL){
sw = sw->left;
}
就这个样子。
递归实现
其实递归实现也非常简单,只用记录上一次的节点指针就行了。
void flatTree(TreeNode * root, TreeNode * head){
if (root == NULL) {
return;
}
flatTree(root->left, head);
if (head == NULL) {
head = root;
}else{
head->right = root;
root->left = head;
head = root;
}
flatTree(root->right, head);
}
自己对代码掌控还是不够熟练的收放自如,还是需要不断的努力啊。