1.问题描述
定义一个函数,输入一个链表的头结点,反转该链表并输出反转后链表的头结点。
2.问题分析
通常链表节点使用一个结构体表示,结构体中包含两个数据:
1、数值
2、指向下一节点的指针
typedef struct listnode{
int data;
struct listnode *next;
}listNode, *list;
反转链表可以采用两种方式
1、使用栈,借助栈的先进后出的特性,重新组织链表,达到反转的目的。
复杂度:空间O(n),时间O(N)。
2、使用三个指针,实现原地反转链表。
复杂度:空间O(1),时间O(N)。
3.代码
#include "stdlib.h"
#include <iostream>
#include <stack>
using namespace std;
typedef struct listnode{
int data;
struct listnode *next;
}listNode, *list;
//使用栈,空间O(n),时间O(N)
list reverseWithStack(list head)
{
if(head == NULL)
return NULL;
stack<list> stk;
list newHead;
list p;
while(head != NULL)
{
stk.push(head);
head = head->next;
}
newHead = stk.top();
p = newHead;
stk.pop();
while(!stk.empty())
{
p->next = stk.top();
stk.pop();
p = p->next;
}
p->next = NULL;
return newHead;
}
//原地反转,空间O(1),时间O(N)
list reverse(list head)
{
list pre = NULL;
list cur = head;
list temp;
while(cur != NULL)
{
temp = cur->next;
cur->next = pre;
pre = cur;
cur = temp;
}
return pre;
}
int main(int argc, char* argv[])
{
//构建测试数据
listNode list1,list2,list3,list4,list5;
list1.data = 1;
list2.data = 2;
list3.data = 3;
list4.data = 4;
list5.data = 5;
list1.next = &list2;
list2.next = &list3;
list3.next = &list4;
list4.next = &list5;
list5.next = NULL;
//list ans = reverseWithStack(&list1); //使用栈
list ans = reverse(&list1); //原地反转
while(ans != NULL)
{
cout << ans->data << endl;
ans = ans->next;
}
system("pause");
return 0;
}
4.总结
推荐原地反转,这样的复杂度可以达到空间O(1),时间O(N)。