牛客刷题二叉树之二叉搜索树与双向链表 and 二叉搜索树的第k个结点

题目描述一二叉搜索树与双向链表

输入一棵二叉搜索树，将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点，只能调整树中结点指针的指向。

牛客链接：

https://www.nowcoder.com/practice/947f6eb80d944a84850b0538bf0ec3a5?tpId=13&&tqId=11179&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

解题思路：

采用两种方式：
第一种为中序遍历的非递归解法；
第二种为剑指offer做法，思路如下：
举例说明：

二叉搜索树如上图所示，我们将其转换为配需双向链表。
根据二叉搜索树的特点：左结点的值<根结点的值<右结点的值，我们不难发现，使用二叉树的中序遍历出来的数据的数序，就是排序的顺序。因此，首先，确定了二叉搜索树的遍历方法。
接下来，我们看下图，我们可以把树分成三个部分：值为10的结点、根结点为6的左子树、根结点为14的右子树。根据排序双向链表的定义，值为10的结点将和它的左子树的最大一个结点链接起来，同时它还将和右子树最小的结点链接起来。

按照中序遍历的顺序，当我们遍历到根结点时，它的左子树已经转换成一个排序的好的双向链表了，并且处在链表中最后一个的结点是当前值最大的结点。我们把值为8的结点和根结点链接起来，10就成了最后一个结点，接着我们就去遍历右子树，并把根结点和右子树中最小的结点链接起来。

代码一：

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
public:
    TreeNode* Convert(TreeNode* pRootOfTree)
    {
        if(pRootOfTree == NULL) return NULL;
        stack<TreeNode*> stk;
        TreeNode* pTemp = pRootOfTree;
        TreeNode* pre = NULL;
        TreeNode* pHead = NULL;
        
        while(pTemp || !stk.empty()) {
            if(pTemp) {
                stk.push(pTemp);
                pTemp = pTemp->left;
            } else {
                pTemp = stk.top();
                stk.pop();
                if (pHead == NULL) {
                    pHead = pTemp;
                    pHead->left = NULL;
                    pre = pHead;
                } else {
                    pTemp->left = pre;
                    pre->right = pTemp;
                    pre = pTemp;
                }
                pTemp = pTemp->right;
            }
        }       
        return pHead;
    }
};

代码二：

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
public:
    TreeNode* Convert(TreeNode* pRootOfTree)
    {
        if(pRootOfTree == NULL) return NULL;
        TreeNode* pLastInList = NULL;
        convertCore(pRootOfTree, &pLastInList);
        
        //遍历找到头结点
        TreeNode* pHead = pLastInList;
        while(pHead != NULL && pHead->left != NULL){
            pHead = pHead->left;
        }
        return pHead;
    }
    
    //遍历整个二叉树，将树节点的左右子树指针变为中序遍历的前后指针
    //pLastInList为已排好序的双向链表最后一个节点
    void convertCore(TreeNode* pRootOfTree, TreeNode** pLastInList) {
        if (pRootOfTree == NULL) return ;
        
        TreeNode* pCurrent = pRootOfTree;
        
        //遍历该节点左子树
        if(pCurrent->left != NULL)
            convertCore(pCurrent->left, pLastInList);
        
        //处理该节点
        pCurrent->left = *pLastInList;
        if(*pLastInList != NULL)
            (*pLastInList)->right = pCurrent;
        
        //将pLastInList变为该节点
        *pLastInList = pCurrent;
        
        //遍历该节点右子树
        if(pCurrent->right != NULL)
            convertCore(pCurrent->right, pLastInList);
    }
};

题目描述二二叉搜索树的第k个结点

给定一棵二叉搜索树，请找出其中的第k小的结点。例如， （5，3，7，2，4，6，8） 中，按结点数值大小顺序第三小结点的值为4。

牛客链接：

https://www.nowcoder.com/practice/ef068f602dde4d28aab2b210e859150a?tpId=13&&tqId=11215&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

代码：

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
public:
    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
        TreeNode* pTemp = pRoot;
        stack<TreeNode*> stk;
        int count = 0;
        while(pTemp || !stk.empty()) {
            if(pTemp) {
                stk.push(pTemp);
                pTemp = pTemp->left;
            } else {
                pTemp = stk.top();
                stk.pop();
                count++;
                if(count == k) {
                    return pTemp;
                }
                pTemp = pTemp->right;
            }
        }
        if(count<k) {
            return NULL;
        }
        return pTemp;
    }  
};

注意：
二叉树中序遍历