题解 | #链表中环的入口结点#

链表中环的入口结点

http://www.nowcoder.com/practice/253d2c59ec3e4bc68da16833f79a38e4

哈希

遍历链表,用map记录链表中每个结点出现的次数,第一次重复出现的结点就是入口节点。

C++代码:

class Solution {
public:
    ListNode* EntryNodeOfLoop(ListNode* pHead) {
        unordered_map<ListNode *, int> mp;
        while (pHead) {
            if (mp[pHead]) {
                return pHead;
            }
            mp[pHead]++;
            pHead = pHead->next;
        }
        return NULL;
    }
};

时间复杂度:O(n) 空间复杂度:O(n)

快慢指针

设置两个指针,快指针每次走两步,慢指针每次走一步,若链表有环,两指针一定会在环中相遇,如下图: alt

设链表起点到入口的距离为a,入口到相遇点距离为b,相遇点到入口的距离为c,因为快指针比慢指针走得快,当慢指针进入环里时,可以看成快指针在环里追赶慢指针,当两者相遇时,快指针至少在环里走过一个完整的圈,否则就说明两个指针速度相等,与假设不符。所以当两指针第一次相遇时,快指针走过的距离rd = a + k(b + c) + b。慢指针走过的距离ld = a + b。因为快指针走过的距离是慢指针的二倍,所以a + k(b + c) + b = 2(a + b),即a = c + (k - 1)(b + c)。这个式子说明,当两指针相遇时,我们把其中一个指针放到起点,另一个指针位置不变,两个指针每次都走一步,他们最终会在起点相遇。留在环里的指针第一次到达入口后,可能会再走(k - 1)圈。

C++代码:

class Solution {
public:
    ListNode* EntryNodeOfLoop(ListNode* pHead) {
        ListNode *fast = pHead, *low = pHead;
        while (fast && fast->next) {
            low = low->next;
            fast = fast->next->next;
            if (fast == low) {break;}
        }
        if (!fast || !fast->next) {return NULL;}
        fast = pHead;
        while (fast != low) {
            fast = fast->next;
            low = low->next;
        }
        return fast;
    }
};

时间复杂度:O(n) 空间复杂度:O(1)

全部评论

相关推荐

Hello_WordN:咱就是说,除了生命其他都是小事,希望面试官平安,希望各位平时也多注意安全
点赞 评论 收藏
分享
09-30 12:39
门头沟学院 C++
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务