只要记住中序遍历的顺序即可,不用开辟数组存储中序的序列,只需判断是否逆序就行了,逆序的用指针标出位置,然后交换个指针的值。想是想不出来的,在纸上画画揪出来了,就像刚才面阿狸的第一道算法题,手头没有笔思路打不开,回来一画揪出来了。如下:
void interTrave(TreeNode *root,TreeNode *&plast,TreeNode *&p,TreeNode *&q,bool &isFirst)
{
if (root==NULL)
{
//*plast = NULL;
return ;
}
if (root->left)
{
interTrave(root->left,plast,p,q,isFirst);
}
if (plast&&plast->val>=root->val)
{
if (isFirst)
{
p = plast;
q = root;
isFirst = false;
}
else
{
q = root;
}
}
plast = root;
if (root->right)
{
interTrave(root->right,plast,p,q,isFirst);
}
}
void recoverTree(TreeNode *root) {
if(root==NULL)
return ;
TreeNode *plast,*p,*q;
plast = NULL;
p = NULL;
q = NULL;
bool isFirst = true;
interTrave(root,plast,p,q,isFirst);
if (p&&q)
{
swap(p->val,q->val);
}
}
做题的时候一直想建立二叉树,想按照层次遍历建立二叉树,真不太容易,先不做。