题干链接:ONT60 旋转链表
这道题是反转链表题的pro升级版,但比反转链表略微复杂一些。如果有做过旋转数组那道题(链接在这里:https://blog.csdn.net/wyd_333/article/details/126712919,但当时刷这道题的时候我用的是二分法,其实是复杂化了),思路很容易想到:先将整个链表整体反转,再分别反转前k个和剩下的。
public ListNode reverse(ListNode head) { ListNode pre = null; ListNode cur = head; while(cur != null) { ListNode curNext = cur.next; cur.next = pre; pre = cur; cur = curNext; } return pre; }
这里反转的这个操作,复用的就是“反转链表”那道题中的思路。这里记录一下AC时容易遇到的几个细节坑:
1. reverse()方法应有返回值。如果调用时候传入的是头节点head,那么作为引用传递,在反转之后head的值是会被改变的,会变成链表的尾部,如果不记录新的头节点,后序就没法再对反转之后的链表进行操作了。可以在调用方法之前记录,也可以通过返回值的方式记录新头节点,推荐返回值的方式。
2.k的值需要处理。在找第k个元素时,k有边界情况要处理。如k > 链表长度。遇到这种情况是直接舍弃,还是轮回,要看题意。这道题题意要求是通过轮回的方式来处理的(WA几次看未通过用例就可以知道了)。因此 k = k % len 是必须的。
3.理清楚谁是头节点,谁是尾节点。这里涉及多次反转,反转之后谁是头谁是尾,要把谁接到谁之后?这些问题是容易出错的,需要想清楚。
代码:
import java.util.*; /* * public class ListNode { * int val; * ListNode next = null; * public ListNode(int val) { * this.val = val; * } * } */ public class Solution { /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param head ListNode类 * @param k int整型 * @return ListNode类 */ public ListNode rotateLinkedList (ListNode head, int k) { if(head == null) { return null; } int len = getLen(head); k = k % len; ListNode afterReverseHead = reverse(head); // testPrint(afterReverseHead); ListNode cur = afterReverseHead; while(k > 1) { k--; cur = cur.next; } System.out.println(cur.val); ListNode newHead = cur.next; cur.next = null; ListNode preHead = reverse(afterReverseHead); ListNode postHead = reverse(newHead); afterReverseHead.next = postHead; return preHead; } public ListNode reverse(ListNode head) { ListNode pre = null; ListNode cur = head; while(cur != null) { ListNode curNext = cur.next; cur.next = pre; pre = cur; cur = curNext; } return pre; } public int getLen(ListNode head) { int count = 0; while(head != null) { head = head.next; count++; } return count; } }