今天做了一道题目,感觉可以记录一下。身为菜鸟的我,在一些边界问题上吃了亏,想了一阵子才处理得较好。
题目:反转链表
定义一个函数,输入一个链表的头结点,反转该链表并输出反转后链表的头结点。
请同时实现迭代版本和递归版本。
样例:
输入:1->2->3->4->5->NULL
输出:5->4->3->2->1->NULL
思想:
(1)可以从头遍历到尾,用数组存起来,然后再从头到尾用数组倒序更新链表,这样比较简单,但需要付出一些空间的代价,并且我们并不知道链表多大,所以申请的空间也不确定。
(2)我们将节点一个一个接到尾巴,如下图。
图中上面是开始的链表,下面是经过反转的链表,序号是步骤。
代码如下:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* reverseList(ListNode* head) {
if(head == NULL) return NULL;
//temp1将当前节点的下一个节点地址存起来,防止当前节点更新next后丢失更新之前的next
//temp2保存的是上一个改变的节点,然后下一个需要改变的节点的next就指向它
//y是当前正在改变的节点。
ListNode* temp1, *temp2 = NULL, *y = head;
while(y != NULL){
temp1 = y -> next;
y -> next = temp2;
temp2 = y;
y = temp1;
}
return temp2;
}
};
然后这道题在乐扣上面有个类似的,稍微改变了一下。题目是指定链表的第left个节点到第right个节点进行反转。
代码如下:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* reverseBetween(ListNode* head, int left, int right) {
//res是寻找第left个节点的前一个节点
ListNode* res = head;
for(int i = 0; i < left - 2; i ++ ){
res = res -> next;
}
ListNode* l = res;
//如果left==1 的话,那么就说明res不存在,l为第left个节点
l = left == 1 ? head : l -> next;
ListNode* r = l;
//r为链表第right个节点
for(int i = 0; i < right - left; i ++ ){
r = r -> next;
}
//将需要反转的范围确定好,这里设置了一个flag就是第left个节点后面的一个节点,用flag保存r->next
//否则在最后的时候r->next会发生改变
ListNode* flag = r->next;
ListNode* temp1, *temp2 = flag, *y = l;
while(y != flag){
temp1 = y -> next;
y -> next = temp2;
temp2 = y;
y = temp1;
}
//如果left==1,那么就说明不存在res(即第left个节点的前一个节点),此时就不用接好了,直接返回temp2
//如果left!=1,那么就说明存在res,此时我们还必须将res->next设置为temp2,将链表接好
if(left == 1) return temp2;
else res -> next = temp2;
return head;
}
};