算法课记录1 - 判断单链表是否是回文

记录一下算法课的代码，欢迎指正。
这篇是判断单链表是不是“回文”。
写了两个函数，各有优势。
头文件是我自己写的，里面是link结点的定义和其他一些函数的声明，不太重要。

简单说一下第一个函数，先用快慢指针找到单链表的中间结点（设其为mid），然后将mid之后的链表逆转，也就是令原来的表尾成为另一个表头，而mid成为表尾。这样我们就有了两个表头（原表头和新表头）和一个表尾（mid）。从两端同时开始遍历单链表并一一比较，若遇到不同的结点则停止遍历并返回false，若其中一个结点到达mid则返回true（返回之前应首先还原链表）。这个函数的优势在于空间复杂度为O(1)。

#include <iostream>
#include <stack>
#include "../Link_functions.h"
using namespace std;

bool isPlaindrome(link* head)
{
	//若只有一个结点，直接返回true
	if (head->next->next == NULL)   
		return true;

	//若只有两个结点，傻瓜式判断是不是回文
	if (head->next->next->next == NULL)
	{
		if (head->next->digit == head->next->next->digit)
			return true;
		else
			return false;
	}
	
	//自己写的函数，获取链表的中间结点，用快慢指针实现的
	link* m = get_mid(head);  

	//定义三个指针，用于逆转后半个链表
	link* a = m;
	link* b = a->next;
	link* c = a->next->next;
	
	//逆转后半个链表，使a、b指向新的首结点
	while (c != NULL)
	{
		b->next = a;
		a = b;
		b = c;
		c = c->next;
	}

	b->next = a;
	a = b;
	
	//令mid成为表尾
	m->next = NULL;
	
	//定义一个指向原链表首结点的指针
	link* p = head->next;

	//两个指针从两端往中间逐一比较，任意一个到达表尾后遍历停止，中途若遇指针不相等的情况则直接返回false
	//懒得重载link的=运算符，所以用digit代替一下
	while (p != m && a != m)
	{
		if (p->digit != a->digit)
			return false;

			p = p->next;
			a = a->next;
	}
	
	//遍历结束后，比较两个指针是否相等，相等则返回true，否则返回false
	//这里需要比较的原因：若链表的结点是偶数个的话，get_mid(head)函数返回的是靠后的那个结点
	//比如说有一个有4个结点的链表，调用get_mid返回的是指向第三个结点的指针
	//这种情况下，后半段链表的指针会先遍历到mid，而此时前半段链表的指针则到达mid的前一个结点，需要将它们再做一次比较来判断整个链表是不是回文
	//返回true或者false前面的这些代码是还原链表
	if (p->digit == a->digit)
	{
		a = b;
		b = a->next;
		c = a->next->next;
		
		a->next = NULL;
		
		while (c != NULL)
		{
			b->next = a;
			a = b;
			b = c;
			c = c->next;
		}

		b->next = a;
	
		return true;
	}
	else
	{ 
		a = b;
		b = a->next;
		c = a->next->next;

		a->next = NULL;

		while (c != NULL)
		{
			b->next = a;
			a = b;
			b = c;
			c = c->next;
		}

		b->next = a;

		return false;
	}
}

这个是用栈判断，很简单

bool isPlaindrome_stack(link* head)
{
	stack<link*> s;

	link* a = head->next;
	link* b = head->next;

	while (a != NULL)
	{
		s.push(a);
		a = a->next;
	}

	while (b != NULL)
	{
		link *temp = s.top();
		s.pop();

		if (b->digit != temp->digit)
			return false;

		b = b->next;
	}

	return true;
}