互联网大厂算法面试题之旋转链表

大家好,我是 程序员小熊,来自某 大厂 的程序猿,今天带来一道来自互联网大厂(字节、腾讯、微软、苹果等) 面试题 Leetcode 61. 旋转链表 ,提供 虚拟头节点 + 双指针 的解题思路,采用 动图 的方式进行层层剖析,供大家参考,希望对大家无论是刷题还是面试都有所帮助。

61. 旋转链表

描述

给你一个链表的头节点 head ,旋转链表,将链表每个节点向右移动 k 个位置。

互联网大厂算法面试题之旋转链表_第1张图片

互联网大厂算法面试题之旋转链表_第2张图片

解题思路

思考

考虑以下几种情况:

特殊情况

  1. 链表为空或只有一个节点;

  2. k 的值为 0 或者为链表长度 L 的整数倍(N > 0)。

一般情况

  1. 链表至少有两个节点且 k 既不等于 0 也不等于 L 的整数倍。

思路

  • 特殊情况

  1. 链表为空或只有一个节点,只要返回头节点即可;

  2. 判断 k == 0,如果等于 0,则直接返回头节点,否则求链表长度 L,再判断 k == NL,如果等于,也可直接返回。

  • 一般情况

  1. 采用 虚拟头节点 + 双指针 的方法,具体如下栗子分析。

举栗

以链表 1->2->3->4->5,k = 2 为栗子,如下图所示。

互联网大厂算法面试题之旋转链表_第3张图片

  • 增加虚拟头节点,设置两根快慢指针 fast/slow 指向它;

互联网大厂算法面试题之旋转链表_第4张图片

  • 先让快指针 fast 向前移动 k 步,慢指针 slow 保持不动;

互联网大厂算法面试题之旋转链表_第5张图片

  • 快慢指针 fast/slow 同时向前移动,直至快指针指向尾节点;

     

    互联网大厂算法面试题之旋转链表_第6张图片

  • 让快指针 fast 指向的节点指向链表的头节点;

互联网大厂算法面试题之旋转链表_第7张图片

  • 将慢指针 slow 指向的节点的下一节点设置为新链表的头节点;

互联网大厂算法面试题之旋转链表_第8张图片

  • 将慢指针 slow 指向的节点设置为新链表的尾节点 slow->next = null;

互联网大厂算法面试题之旋转链表_第9张图片

  • 旋转原链表后得到新链表。

互联网大厂算法面试题之旋转链表_第10张图片

完整动图

互联网大厂算法面试题之旋转链表_第11张图片

反思

  1. 如何找到旋转之后得到的新链表的头节点?

采用 双指针 中的 快慢指针

  1. 设置虚拟头节点的作用?

本题设置 虚拟头节点 的作用是找到 新链表的头节点和尾节点。通过 快慢指针 去寻找。

Show me the Code

C++

ListNode* rotateRight(ListNode* head, int k) {
    /*  特殊情况判断  */
    if (head == nullptr || head->next == nullptr || k == 0) {
        return head;
    }

    /*  增加虚拟头节点  */
    ListNode* dummy = new ListNode(0);
    dummy->next = head;

    /*  获取链表长度  */
    int len = 0;
    for (ListNode* node = head; node != nullptr; node = node->next) {
        len++;
    }

    k %= len;
    /*  判断 k 是否是 len 的整数倍  */
    if (k == 0) {
        return head;
    }

    /*  获取新链表的头尾节点  */
    ListNode *fast = dummy, *slow = dummy;
    for (int i = 0; i < k; ++i) {
        fast = fast->next;
    } 
    while (fast->next != nullptr) {
        fast = fast->next;
        slow = slow->next;   
    }        
    fast->next = head; 
    head = slow->next; 
    slow->next = nullptr;

    /*  释放虚拟头节点  */
    delete dummy;
    
    return head;
}

java

ListNode rotateRight(ListNode head, int k) {
    if (k == 0 || head == null || head.next == null) {
        return head;
    }

    ListNode dummy = new ListNode(0);
    dummy.next = head;

    int len = 0;
    for (ListNode node = head; node != null; node = node.next) {
        len++;
    }     

    k %= len;
    if (k == 0) {
        return head;
    }

    ListNode fast = dummy, slow = dummy;
    for (int i = 0; i < k; ++i) {
        fast = fast.next;
    } 
    while (fast.next != null) {
        fast = fast.next;
        slow = slow.next;   
    }        
    fast.next = head; 
    head = slow.next; 
    slow.next = null;

    return head;
}

往期链表相关文章精彩回顾

动图:删除链表的倒数第 N 个结点

虚拟头节点秒杀链表问题

单链表之环形链表

四种方法解决leetcode203. 移除链表元素

字节面试题 leetcode 83. 删除排序链表中的重复元素

面试不可不会的单链表反转

更多精彩

关注公众号【程序员小熊】,后台回复【算法】、【python】或【1024】,即可获取高清无码的经典算法、 python 电子书或三份来自 BAT 大佬的 Leetcode 刷题手册~

互联网大厂算法面试题之旋转链表_第12张图片

为了回馈读者,本公众号不定期会有送礼活动,敬请关注~

你可能感兴趣的:(刷题,面试,链表,链表,算法,leetcode,面试)