剑指 Offer-36 二叉搜索树与双向链表c++

题目描述

解法 dfs

很容易发现这题就是考察二叉搜素树的中序遍历是递增序列的性质。
我们先写出中序遍历的代码

void dfs(Node* root) {
	if(root == NULL) return;
	dfs(root -> left);
	cout << root -> val;
	dfs(root -> right);
}

可以发现只要对中间的主体做出改动，使遍历过程中的结点指针指向对应位置就能形成双向链表。
但题目还要求是循环链表，所以我们需要记录头结点和尾结点的位置。
思路：
连接部分：设置pre结点，初始值为空，dfs函数的参数为cur结点。递归过程中

		pre -> right = cur;
        cur -> left = pre;
        pre = cur;

头尾结点部分：
设置局部变量head，调用dfs初次进入递归的中间部分时，说明我们到了头结点，于是head = cur。

		if(pre == NULL) head = cur; 
		else pre -> right = cur;
	    cur -> left = pre;
        pre = cur;

作为递归函数的上一个结点，当cur为空时（即递归结束时），pre就是尾结点。

 		dfs(root);
		pre -> right = head;
 		head -> left = pre;

于是得到完整代码

class Solution {
public:
    vector<Node*> ans;
    Node* treeToDoublyList(Node* root) {
        if(root == NULL) return nullptr;
        dfs(root);
        int len = ans.size();
        for(int i = 0; i <= len -1 ; i ++) {
            ans[i] -> left = ans[(i - 1 + len) % len];
            ans[i] -> right = ans[(i + 1) % len ];
        }
        return ans[0];
    }
    void dfs(Node* root) {
        if(root == NULL) return;
        dfs(root -> left);
        ans.push_back(root);
        dfs(root -> right);
    }
};

这里头尾结点的相连使用了经典循环链表的连法，
当i = 0 时，ans[(i - 1 + len) % len]为ans[len - 1] 指向尾元素元素
当i = len -1 时,ans[(i + 1) % len]为ans[0] 指向首元素
时间复杂度O(N)，空间复杂度O(N)