程序员面试经典(9)：环路检测

题目描述

给定一个有环链表，实现一个算法返回环路的开头节点。
有环链表的定义：在链表中某个节点的next元素指向在它前面出现过的节点，则表明该链表存在环路。

我的解题

快慢指针

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode *detectCycle(ListNode *head) {
        if(head==nullptr || head->next==nullptr) return nullptr;
        ListNode *p=head, *q=head->next->next;
        while(p!=q)
        {
            if(q==nullptr) return nullptr;
            p=p->next;
            if(q==nullptr) return nullptr;
            else q=q->next;
        }
        
        int count=1;
        p=p->next;

        while(p!=q)
        {
            count++;
            p=p->next;
        }

        p=head;q=head;
        while(count-->0) q=q->next;
        
        while(p!=q)
        {
            p=p->next;q=q->next;
        }
        return p;
    }
};

执行用时 :12 ms, 在所有 C++ 提交中击败了80.00%的用户
内存消耗 :10 MB, 在所有 C++ 提交中击败了100.00%的用户