leetcode之Rotate List

原题如下:

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;
    }
};


发布了85 篇原创文章 · 获赞 0 · 访问量 2万+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章