力扣做题记录——算法入门day5
876.链表的中间结点
给定一个头结点为head
的非空单链表,返回链表的中间结点。如果有两个中间结点,则返回第二个中间结点。
示例1
输入:[1,2,3,4,5]
输出:此列表中的结点 3 (序列化形式:[3,4,5])
返回的结点值为 3 。 (测评系统对该结点序列化表述是 [3,4,5])。
注意,我们返回了一个 ListNode 类型的对象 ans,这样:
ans.val = 3, ans.next.val = 4, ans.next.next.val = 5, 以及 ans.next.next.next = NULL.
示例2
输入:[1,2,3,4,5,6]
输出:此列表中的结点 4 (序列化形式:[4,5,6])
由于该列表有两个中间结点,值分别为 3 和 4,我们返回第二个结点。
提示
- 给定链表的结点数介于
1
和100
之间。
这道题可以使用快慢指针的思想,快指针每次移动两个结点,慢指针每次移动一个结点,这样当快指针移动到表尾时,慢指针正好位于中间结点。
代码
ListNode* middleNode(ListNode* head) {
ListNode *slow = head;
ListNode *fast = head;
while (fast && fast->next)
{
slow = slow->next;
fast = fast->next->next;
}
return slow;
}
要想让慢指针指向第一个中间结点,就在判断条件中加入fast->next->next != NULL。
此外还有数组法和单指针法,代码如下,不再解释:
// 数组法
ListNode* middleNode(ListNode* head) {
vector<ListNode*> A = {head};
while (A.back()->next != NULL)
A.push_back(A.back()->next);
return A[A.size() / 2];
}
// 单指针法
ListNode* middleNode(ListNode* head) {
int n = 0;
ListNode* cur = head;
while (cur != nullptr) {
++n;
cur = cur->next;
}
int k = 0;
cur = head;
while (k < n / 2) {
++k;
cur = cur->next;
}
return cur;
}
19.删除链表的倒数第N个结点
给你一个链表,删除链表的倒数第n
个结点,并且返回链表的头结点。
示例1
输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]```
示例2
输入:head = [1], n = 1
输出:[]
示例3
输入:head = [1,2], n = 1
输出:[1]
提示
- 链表中结点数目为
sz
1 <= sz <= 30
0 <= Node.val <= 100
1 <= n <= sz
还是快慢指针,第一遍先把快指针移到超前慢指针n的位置,再将两指针一起后移直到快指针到达表尾。然后将慢指针的结点指向下下个结点,返回头指针。
代码
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* dummy = new ListNode(0, head);
stack<ListNode*> stk;
ListNode* cur = dummy;
while (cur) {
stk.push(cur);
cur = cur->next;
}
for (int i = 0; i < n; ++i) {
stk.pop();
}
ListNode* prev = stk.top();
prev->next = prev->next->next;
ListNode* ans = dummy->next;
delete dummy;
return ans;
}
或者遍历两次,第一次记录链表长度,第二次删除倒数第n个结点。或者用栈,将结点依次入栈,弹出栈的第n个结点就是要删除结点的前驱结点。
本文题目及部分答题思路来自Leetcode。
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/middle-of-the-linked-list/solution/lian-biao-de-zhong-jian-jie-dian-by-leetcode-solut/
来源:力扣(LeetCode)