leetcode之Rotate List

来源:互联网 发布:淘宝做话费充值 编辑:程序博客网 时间:2024/05/21 22:54

原题如下:

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.

这道题关键是理解题意,这里是循环旋转的,所以k可以无限大,但k与k%n之后的旋转效果是相同的详见:https://oj.leetcode.com/discuss/2353/what-to-do-when-k-is-greater-than-size-of-list。理解了题意之后编码还是比较容易实现的,首先遍历一遍链表求出其长度,同时将tail指针指向最后一个节点(所以此时求长度是从1开始的)。另外可以发现,当k% n等于0时,只需返回原链表,否则,利用指针p和q分别指向要断开链表的地方(即p指向第n - k个节点,q指向第n-k +1个节点),注意,由于第一个节点的下标为1,所以在遍历时要注意与数组遍历时的不同,确保指针指向正确的位置。

class Solution {public:    ListNode *rotateRight(ListNode *head, int k) {        if(head == NULL || k == 0)return head;int n = 1;ListNode * p = head, *q = head, *tail = head;while(tail->next != NULL){    n++;tail = tail->next;}k = k % n;if(k == 0)return head;for(int i = 1; i < n - k; i++){    p = p->next;}q = p->next;p->next = NULL;tail->next = head;return q;    }};


0 0
原创粉丝点击