题干链接: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;
}
}