LeetCode: 将链表中的索引为m到n的m-n+1个数逆置,输出该链表(c++实现)
举例说明
假设现在有一个链表:1->2->3->4->5,现在要求将索引 2 到 4 的三个数,也就是1 2 3逆序,使得最终的输出为 1->4->3->2->5.
源代码
Note: 代码中的head表示头节点,指向的是第一个节点(也就是例子中的1),而不是指向空结点。
#include<iostream>
using namespace std;
// 创建了一个ListNode类来表示结点(由数值和next指针构成)
class ListNode
{
public:
int val;
ListNode* next;
ListNode(int v):val(v){}
};
//用来遍历输出一个链表,输入的参数为链表的头指针
void Traversal_List(ListNode* head)
{
while(head)
{
cout << head->val << ",";
head = head->next;
}
}
class Solution
{
// **********核心代码************
// 需要输入的参数为链表的头节点,以及要逆置的结点的起点(m)和终点(n)
// 下面的代码是以m=2 ,n =4 来举例说明的!!!!
ListNode* Reverse_List_m_n(ListNode* head, int m, int n)
{
// 先创建一个pre_head指针,用来存放第m-1个结点(结合例子来看,就是指向数值1的指针)
// 但我们现在还没找到第m个结点呢,所以先把pre_head置为NULL
ListNode* pre_head = NULL;
// 创建一个result指针来指向头结点(如果不发生特殊情况的话,比如例子中我们只逆序了2到4,并
//没有对头结点做改动,那只需要返回result指针(头结点)就行了。)
ListNode* result = head;
// 记录一下要逆序的结点的个数
int reverse_len = n-m + 1;
//这个while循环的目的就是找到第m个结点,也就是要逆置的第一个数
// 退出循环时,head指向第m个结点,pre_head指向第m-1个结点
// 结合例子来看, 就是分别指向了数字2 和数字 1.
while (head && --m)
{
pre_head = head;
head = head->next;
}
//创建了mnList_tail指针来指向head
//(此时的head指向的是第m个结点,也就是要逆置的mn序列的'头结点')
// 至于为什么这个指针的名字取为_tail,是因为最终mnList——tail会作为这个mn序列的'尾部' \\
与n+1个结点相连
ListNode* mnList_tail = head;
//接下来的代码到while循环结束, 实现的是将第m到第n之间的数逆置
//关于这个while循环就不具体讲述了...
ListNode* new_head = NULL;
while (head && reverse_len--)
{
ListNode* next = head->next;
head->next = new_head;
new_head = head;
head = next;
}
// while循环结束之后,此时new_head指向的是第n个结点,也就是例子中的数字4,同时head指向的是
// 第n+1个结点,也就是例子中的数字5.\\
//(这里刚好第n+1个结点是尾结点,但一般第n+1个结点不是尾结点)
//接下来就是把这个逆置的mn序列,与第m-1个结点和第n+1个结点首尾相连了。具体操作如下:
//指针mnList_tail应该与第n+1个结点(也就是当前的head)相连,也就是例子中的数字2 \\
//与数字5 相连。
mnList_tail->next = head;
//接下来特殊情况就来了,即pre_head非空吗??
//回过头看第一个while循环,如果m为1,那么--m为0,此时while循环就不会执行,pre_head仍为\\
//一开始的NULL。那么m = 1表示的意思就是要逆置的序列的第一个数就是头结点。
if (pre_head) #pre_head非空,i.e. m>1
{
// 将第m-1个结点(pre_head)指向第n个结点(也就是逆序的mn序列的最后一个结点,i.e. \\
//new_head ) 结合例子来看,就是把数字1指向了数字4.
pre_head->next = new_head;
}
else
{
//否则,m = 1,即要逆序的起点是链表的头结点,也就是数字1.
//那么此时,我需要返回的新链表的头结点result指向的不再为数字1了(result的初始值\\
//一开始指向的是数字1),而是mn序列的尾结点,也就是数字4,也就是当前new_head指向的结点。
result = new_head;
}
return result;
}
};
代码测试:
int main()
{
ListNode a1 = ListNode(1);
ListNode a2 = ListNode(2);
ListNode a3 = ListNode(3);
ListNode a4 = ListNode(4);
ListNode a5 = ListNode(5);
a1.next = &a2;
a2.next = &a3;
a3.next = &a4;
a4.next = &a5;
cout << "正常输出:" << endl;
Traversal_List(&a1);
cout << endl;
Solution s1;
ListNode* new_head = s1.Reverse_List_m_n(&a1,2,4);
cout << "逆序idx为2到4的数值" << endl;
Traversal_List(new_head);
}