題目
Given a list, rotate the list to the right by k places, where k is non-negative.
For example:
Given 1->2->3->4->5->NULL and k = 2,
return 4->5->1->2->3->NULL.
分析
鏈表操作谨垃,沒什么好說的。需要注意的是對(duì)于空鏈表的處理以及k大于鏈表長(zhǎng)度的處理钝尸。
實(shí)現(xiàn)
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* rotateRight(ListNode* head, int k) {
int length=0;
for(ListNode *p=head; p!=NULL; p=p->next){
length++;
}
if(length==0) return head;
k = k%length;
ListNode *p1=head, *p2=head;
while(k--) p2 = p2->next;
while(p2->next != NULL){
p1 = p1->next;
p2 = p2->next;
}
p2->next = head;
head = p1->next;
p1->next = NULL;
return head;
}
};
思考
LeetCode上對(duì)于邊界輸入的要求還是比較高的筑公,這方面可以多注意取胎。