【数据结构-链表】旋转链表

mac2023-02-02  23

1.题目61. 旋转链表

给定一个链表,旋转链表,将链表每个节点向右移动 k 个位置,其中 k 是非负数。 输入: 1->2->3->4->5->NULL, k = 2 输出: 4->5->1->2->3->NULL 解释: 向右旋转 1 步: 5->1->2->3->4->NULL 向右旋转 2 步: 4->5->1->2->3->NULL 将链表分成段,第一段是k%len长,是链表的后k%len个元素 需要到队头 第二段是len - k%len 是链表的前面几个元素 需要放到第一段的后面

2.思路

算法: * 1.先计算链表长度len * 2.记录旧队头oldHead,先走len - k%len,找到新队头newHead * 3.记得断开新队尾 * 4.走到队尾,tail.next=oldHead * 5.返回新队头

3.代码实现

public ListNode rotateRight(ListNode head, int k) { if(head==null) return null; int len = getLen(head); int secLen = len - k%len; ListNode oldHead = head; for (int i=0;i<secLen-1;i++) { head = head.next; } ListNode newHead = head.next; head.next = null; //断开新队尾 head = newHead; // 找到链尾 while (head!=null && head.next!=null) { head = head.next; } if (head!=null){ head.next = oldHead; } return newHead!=null?newHead:oldHead; } private int getLen(ListNode head) { int cnt = 0; while (head!=null) { cnt++; head = head.next; } return cnt; }

4.注意点

1.边界问题 当k=0或者k=len时,新队头为空,记得返回旧队头

2.一定要记得衔接断开,否则一定是死循环

最新回复(0)