查看原文
其他

【链表问题】打卡9:将单链表的每K个节点之间逆序

帅地 帅地玩编程 2022-06-18

前言

以专题的形式更新刷题贴,欢迎跟我一起学习刷题,相信我,你的坚持,绝对会有意想不到的收获。每道题会提供简单的解答,如果你有更优雅的做法,欢迎提供指点,谢谢。

注:如果代码排版出现了问题麻烦通知我下,谢谢。

【题目描述】

给定一个单链表的头节点head, 实现一个调整单链表的函数,使得每K个节点之间逆序,如果最后不够K个节点一组,则不调整最后几个节点。

例如:

链表:1->2->3->4->5->6->7->8->null, K = 3。

调整后:3->2->1->6->5->4->7->8->null。其中 7,8不调整,因为不够一组。

【要求】

如果链表的长度为 N, 时间复杂度达到 O(N)。

【难度】

尉:★★☆☆

【解答】

对于这道题,如果你不知道怎么逆序一个单链表,那么可以看一下我之前写的【链表问题】如何优雅着反转单链表

这道题我们可以用递归来实现,假设方法reverseKNode()的功能是将单链表的每K个节点之间逆序。reverse()方法的功能是将一个单链表逆序。

那么对于下面的这个单链表,其中 K = 3。

我们把前K个节点与后面的节点分割出来:

temp指向的剩余的链表,可以说是原问题的一个子问题。我们可以调用reverseKNode()方法将temp指向的链表每K个节点之间进行逆序。再调用reverse()方法把head指向的那3个节点进行逆序,结果如下:

接着,我们只需要把这两部分给连接起来就可以了。最后的结果如下:

如果不大理解,看下代码可能就比较好理解了。

代码如下

1//每k个节点为一组的逆转
2public static Node reverseKNodes(Node head, int k) {
3    if (head == null || head.next == null) {
4        return head;
5    }
6    Node cur = head;
7    for (int i = 1; cur != null && i < k; i++) {
8        cur = cur.next;
9    }
10    //判断是否能组成一组。
11    if (cur == null) {
12        return head;
13    }
14    //temp指向剩余的链表
15    Node temp = cur.next;
16    cur.next = null;
17    //把k个节点进行反转
18    Node newHead = reverse(head);
19    //把之后的部分链表进行每K个节点逆转转
20    Node newTemp = reverseKNodes(temp, k);
21    //把两部分节点连接起来
22    return newHead;
23}
24
25    //单链表逆序
26public static Node reverse(Node head) {
27    if (head == null || head.next == null) {
28        return head;
29    }
30    Node newHead = reverse(head.next);
31    head.next.next =  head;
32    head.next = null;
33    return newHead;
34}

当然,这道题一个很简单的做法就是利用栈来辅助,每K个节点入栈就把这K个节点出栈连接成一个链表,之后剩余再在进栈…..

不过这种做法的额外空间复杂度是O(K)。

问题拓展

思考:如果这是一个环形单链表呢?该如何实现呢?

往期

【链表问题】打卡8:复制含有随机指针节点的链表

【链表问题】打卡7:将单向链表按某值划分成左边小,中间相等,右边大的形式

为了方便寻找文章,我把文章目录整理了下,把文章导航放在了github上:https://github.com/iamshuaidi/writing


您可能也对以下帖子感兴趣

文章有问题?点此查看未经处理的缓存