我这题写得像老奶奶缝补臭袜子。
最开始通过了3%的样例,是因为没有判断head==nullptr
,写了之后通过了6%。后来又把k=k%len
加上去了,这回才AC。
这题简单粗暴,按照题目上说的逻辑来拆分组合链表就好了,要注意的就是异常情况的处理。
牛客网上的数据头一个就是判断对最简单异常情况的处理,每次发生这种错误的时候就要提醒自己,提交前异常情况处理了吗?都考虑周到了吗?毕竟以后面试笔试的时候可没人会让我一遍一遍的改错。
class Solution {
public:
ListNode *rotateRight(ListNode *head, int k) {
if (head == nullptr) return nullptr;
if (head->next == nullptr) return head;
ListNode *plen = head;
int len = 0;
while (plen != nullptr) { plen = plen->next; len++; }
ListNode *newHead = head, *fast = head;
if (k == len) return head;
k = k % len;
while (k-- && fast->next != nullptr) fast = fast->next;
while (fast->next != nullptr) {
fast = fast->next;
newHead = newHead->next;
}
fast->next = head;
head = newHead->next;
newHead->next = nullptr;
return head;
}
};