首页 > 代码库 > 如何检查一个单向链表上是否有环?

如何检查一个单向链表上是否有环?

1, 最简单的方法, 用一个指针遍历链表, 每遇到一个节点就把他的内存地址(java中可以用object.hashcode())做为key放在一个hashtable中. 这样当hashtable中出现重复key的时候说明此链表上有环. 这个方法的时间复杂度为O(n), 空间同样为O(n).

2, 使用反转指针的方法, 每过一个节点就把该节点的指针反向:

Boolean reverse(Node *head) {
    Node *curr = head;
    Node *next = head->next;
    curr->next = NULL;

    while(next!=NULL) {
        if(next == head) { /* go back to the head of the list, so there is a loop */
            next->next = curr;
            return TRUE;

        }

        Node *temp = curr;
        curr = next;
        next = next->next;
        curr->next = temp;
    }

    /* at the end of list, so there is no loop, let‘s reverse the list back */
    next = curr->next;
    curr ->next = NULL;
    while(next!=NULL) {
        Node *temp = curr;
        curr = next;
        next = next->next;
        curr->next = temp;
    }
    return FALSE;
}

看上去这是一种奇怪的方法: 当有环的时候反转next指针会最终走到链表头部; 当没有环的时候反转next指针会破坏链表结构(使链表反向), 所以需要最后把链表再反向一次. 这种方法的空间复杂度是O(1), 实事上我们使用了3个额外指针;而时间复杂度是O(n), 我们最多2次遍历整个链表(当链表中没有环的时候).

这个方法的最大缺点是在多线程情况下不安全, 当多个线程都在读这个链表的时候, 检查环的线程会改变链表的状态, 虽然最后我们恢复了链表本身的结构, 但是不能保证其他线程能得到正确的结果.

3, 这是一般面试官所预期的答案: 快指针和慢指针

Boolean has_loop(Node *head) {
    Node *pf = head; /* fast pointer */
    Node *ps = head; /* slow pointer */

    while(true) {
        if(pf && pf->next)
            pf = pf->next->next;
        else
            return FALSE;

        ps = ps->next;

        if(ps == pf)
            return TRUE;
    }
}

需要说明的是, 当慢指针(ps)进入环之后, 最多会走n-1步就能和快指针(pf)相遇, 其中n是环的长度. 也就是说快指针在环能不会跳过慢指针, 这个性质可以简单的用归纳法来证明.
(1)当ps在环中位置i, 而pf在环中位置i-1, 则在下一个iteration, ps会和pf在i+1相遇.
(2)当ps在环中位置i, 而pf在环中位置i-2, 则在下一个iteration, ps在i+1, pf在i, 于是在下一个iteration ps和pf会相遇在i+2位置
(3)和上面推理过程类似, 当ps在i, pf在i+1, 则他们会经过n-1个iteration在i+n-1的位置相遇. 于是慢指针的步数不会超过n-1.

扩展:

这个问题还有一些扩展, 例如, 如何找到环的开始节点? 如何解开这个环? 这些问题的本质就是如何找到有"回边"的那个节点.

我们可以利用方法3的一个变形来解决这个问题:

Boolean has_loop(Node *head) {
    Node *pf = head; /* fast pointer */
    Node *ps = head; /* slow pointer */

    while(true) {  /* step 1, is there a loop? */
        if(pf && pf->next)
            pf = pf->next->next;
        else
            return FALSE;

        ps = ps->next;

        if(ps == pf)
            break;
    }

    /* step 2, how long is the loop */
    int i = 0;
    do {
        ps = ps->next;
        pf = pf->next->next;
        i++;
    } while(ps!=pf)

    /* step 3, use 2 addtional pointers with distance i to break the loop */
    ps = head;
    pf = head;

    int j;
    for(j=0; j<i; j++) { pf = pf->next; }

    j = 0;
    while(ps!=pf) {
        ps = ps->next;
        pf = pf->next;
        j++;
    }

    printf("loop begins at position %d, node address is %x", j, ps);

    /*step 4, break the loop*/
    for(j=0; j<=i; j++) { ps = ps->next; } //step i-1 in the loop from loop beginning
    ps->next = NULL; // break the loop
    return TRUE;
}

如何检查一个单向链表上是否有环?