若链表共n个结点,观察到新链表的最后一个结点就是原来链表第n-k%n个结点
我们可以先将给定的链表连接成环,然后将指定位置断开。
我们首先计算出链表的长度 n,并找到该链表的末尾节点,将其与头节点相连。这样就得到了闭合为环的链表。
然后我们找到新链表的最后一个节点,并记录新链表头结点,再将当前闭合为环的链表断开,即可得到我们所需要的结果。
特别地,当链表长度不大于 1,或者 k为 n 的倍数时,新链表将与原链表相同,我们无需进行任何处理。
class Solution {
public:
ListNode* rotateRight(ListNode* head, int k) {
if(!head||!head->next|k==0){return head;}//0个或1个结点或k为0,不必移动直接返回
ListNode*p=head;int n=1;
while(p->next){n++;p=p->next;}//计算结点个数
p->next=head;//把链表连成环
int add=n-k%n;//计算要断开的位置
while(add--){
p=p->next;
}
ListNode*q=p->next;//记录新的头结点
p->next=nullptr;
return q;
}
};
下面为力扣官方题解,add这里可以优化一下。
class Solution {//来源:力扣官方题解
public:
ListNode* rotateRight(ListNode* head, int k) {
if (k == 0 || head == nullptr || head->next == nullptr) {
return head;
}
int n = 1;
ListNode* iter = head;
while (iter->next != nullptr) {
iter = iter->next;
n++;
}
int add = n - k % n;
if (add == n) {//优化:当add=n说明k是n的整数倍,不需移动
return head;
}
iter->next = head;
while (add--) {
iter = iter->next;
}
ListNode* ret = iter->next;
iter->next = nullptr;
return ret;
}
};