代码随想录打卡5
今日题目:
两两交换链表中的节点
题目:
给你一个链表,两两交换其中相邻的节点,并返回交换后链表的头节点。你必须在不修改节点内部的值的情况下完成本题(即,只能进行节点交换)。
思路:
本题需要用到三个指针,pre和p两个指针负责结点交换,指针temp指向需要交换的两个元素之前的结点,即上图中3和4交换,temp指向结点1。因为在结点3和结点4交换后,结点1的next域仍然指向3结点,因此需要更新1的指针域,使其指向结点4。
如果不使用temp记录,则上图交换后的结果会变为[2,1,3]。
下面讲一下交换结点的步骤,这里pre指向第一个结点,p指向第二个。
首先将结点1的next指向3,即pre->next = p->next;
然后将结点2的next指向结点1,即p->next = pre;
因为起始结点1之前没有结点,因此不需要temp指向,为了后面3、4结点的交换,需要temp记录结点1的位置,即temp = pre;
。
最后更新p、pre位置。
当进入第二次循环时,需要用记录的temp指针域指向后面交换好的结点,即temp->next = p;
在编写代码时,需要注意head结点的位置,题目的head结点即为第一个元素,在1、2元素交换位置后,第一个结点元素变为2号结点,所以需要更新一次head,使其指向新的头结点2。
最后需要注意的是对空指针的使用,这是写链表题目经常出错的一个地方。因此需要增加一些判定条件,确保使用的指针非空。
附上代码:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* struct ListNode *next;
* };
*/
struct ListNode* swapPairs(struct ListNode* head){
if(head == NULL || head->next == NULL){
return head;
}
struct ListNode *pre = head;
struct ListNode *p = head->next;
head = p;//注意head的位置,提前更改好
struct ListNode *temp = NULL;
/*
错误原因:试图使用空指针
解决方法:增加判断条件,并且判断的顺序不能改变。排除对空指针的引用。
*/
while(pre && p){//应该多加一个判定条件 p也不空,不然下一行代码会报错
pre->next = p->next;//error:试图使用空指针
p->next = pre;
if(temp)//这里第一次循环进入不需要执行
temp->next = p;
temp =pre;
pre = pre->next;
if(pre != NULL)//避免使用空指针
p = pre->next;
}
return head;
}
删除链表的倒数第 N 个结点
题目:
给你一个链表,删除链表的倒数第 n 个结点,并且返回链表的头结点。
示例 :
输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]
思路:应用了快慢指针的思想。
双指针的经典应用,如果要删除倒数第n个节点,让fast移动n步,然后让fast和slow同时移动,直到fast指向链表末尾。删掉slow所指向的节点就可以了。
编写代码时需要注意虚拟头结点的使用方法。
先定义一个虚拟头结点s,使其指向head,在完成结点删除并返回链表时,再更新head的位置,即 head = s->next;
。可能有人会觉得,在正常删除结点的整个过程中head的位置是没有变化的,在删除完成后直接返回head即可,这句代码完全是多余的。其实不然,本题需要考虑特殊情况,当head只有一个结点且删除倒数第一个结点(即head头结点)时,经过操作后head结点是没有被删除的,直接返回head会出错。
本人代码由于开始没有虚拟结点经常报错,看了随想录代码后才使用的虚拟头结点,代码有点乱,在此直接引用随想录的示例代码。
/**c语言单链表的定义
* Definition for singly-linked list.
* struct ListNode {
* int val;
* struct ListNode *next;
* };
*/
struct ListNode* removeNthFromEnd(struct ListNode* head, int n) {
//定义虚拟头节点dummy 并初始化使其指向head
struct ListNode* dummy = malloc(sizeof(struct ListNode));
dummy->val = 0;
dummy->next = head;
//定义 fast slow 双指针
struct ListNode* fast = head;
struct ListNode* slow = dummy;
for (int i = 0; i < n; ++i) {
fast = fast->next;
}
while (fast) {
fast = fast->next;
slow = slow->next;
}
slow->next = slow->next->next;//删除倒数第n个节点
head = dummy->next;
free(dummy);//删除虚拟节点dummy
return head;
}