LeetCode 876. 链表的中间结点 C++

题目描述

给定一个头结点为 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 之间。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/middle-of-the-linked-list

解题思路

1.首先两种情况特判：
如果链表为空，或者链表中只有一个元素的话，我们就直接返回当前链表。
2.否则的话，我们就先计数，算出链表的长度sz。
3.然后再继续从头开始，找到第sz/2+1的位置，就是我们要返回的中间结点

代码

/**
 * 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* middleNode(ListNode* head) {
        if(head==NULL||head->next==NULL){//两种情况特判：链表为空或者链表中只有一个结点
            return head;//直接返回当前结点
        }
        int sz=0;//sz用来记录数组的长度
        ListNode* p=head;//指针p用来扫描数组
        while(p!=NULL){//当前结点非空时候
            sz++;//计数器加1
            if(p->next==NULL){//如果下一个结点为空结点的话，说明遍历结束了直接跳出循环
                break;
            }
            p=p->next;//否则的话，移动指针后移
        }
        p=head;//遍历结束之后让p再次指向原来链表的头结点，即指向了位置为1的元素
        int i=1;//我们从位置为1的元素后面开始扫描，因为没有设置头结点所以就用i=1来扫了。
        while(i<sz/2+1){//当小于目标位置的时候
            p=p->next;//指针后移
            i++;
        }
        return p;//最后返回p结点即可
    }
};