Leetcode 426 && JZ36 二叉搜索树与双向链表

• 先构造一个head pre指针
  • head 不用说，是找根结点
  • pre有两个作用
    • 找整个排序数组的起点
    • 进行穿针引线时，用来调用上一层函数中的值，即：利用它来找到上一层中的某个结点。否则的话，如果都在当前函数里传指针，会出现「自己指向自己」，无限递归。错误答案见最后
• 每次需要想用flag的时候，再想想看有没有其它的代替方案，会不会那个flag还有其它的很明显的改变
  比如 pre==null 就相当于找到了
• 刚开始想到中序遍历很好，紧接着没有新的idea。如果这时候，顺着中序遍历走一遍，可能就会发现要实现目标功能，要修改的东西很少很简单

/*
// Definition for a Node.
class Node {
public:
    int val;
    Node* left;
    Node* right;

    Node() {}

    Node(int _val) {
        val = _val;
        left = NULL;
        right = NULL;
    }

    Node(int _val, Node* _left, Node* _right) {
        val = _val;
        left = _left;
        right = _right;
    }
};

*/
class Solution {
public:
    Node *head, *pre;
    Node *treeToDoublyList(Node *root) {
        if(!root) return nullptr;
        dfs(root);
        head->left = pre;
        pre->right = head;
        return head;
    }
    void dfs(Node* cur){
        if(cur==nullptr) return ;
        dfs(cur->left);
        if(pre==nullptr) head=cur;
        else pre->right=cur; 
        cur->left = pre;
        pre = cur ;
        dfs(cur->right);
    }
};

错误答案：无限递归

class Solution {
public:
    Node* treeToDoublyList(Node* root) {
        dfs(root);
        Node* cur = head;
        while(!cur->right){
            cur=cur->right;
        }
        cur->right = head;
        head->left = cur;
        return head;
    }
    void dfs(Node* root){
//        cout<<root->val<<" ";
        if(root==NULL) return ;
        dfs(root->left);
        Node* left = root->left;
        Node* right = root->right;
        if(flag && left==NULL && right==NULL) {
            head = root; flag = false;
        }
        if(left) left->right = root;
        if(right) right->left = root;
        dfs(root->right);
    }
private:
    Node* head;
    bool flag = true;
};