码迷,mamicode.com
首页 > 其他好文 > 详细

[LeetCode]Rotate List

时间:2015-03-20 23:53:07      阅读:287      评论:0      收藏:0      [点我收藏+]

标签:

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大于或等于链表的长度n,所以实际效果等同于循环右移k%n次。

  1. 遍历单链表,计算出其长度n,并保存指向最后一个结点的尾指针tail
  2. 对k做预处理,k = k % n
  3. 顺序遍历单链表,定位到第n-k个结点
  4. 保存第n-k个结点的后继结点p,将tail的next置为head,并将原来的Head置为q,同时将第n-k个结点的next置为NULL。

需要注意的空链表和n==k的情况。

下面贴上代码:

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* getLength(ListNode* head,int& len){
        ListNode* p = head;
        len = 1;
        while (p&&p->next){
            len++;
            p = p->next;
        }
        return p;
    }

    ListNode *rotateRight(ListNode *head, int k) {
        int len;
        if (head){
            ListNode* tail = getLength(head, len);
            k = k%len;
            int count = len - k;
            ListNode* p = head;
            while (p&&count > 1){
                p = p->next;
                count--;
            }
            ListNode* q = p->next;
            p->next = NULL;
            if (q){
                tail->next = head;
                head = q;
            }
        }
        return head;
    }
};

[LeetCode]Rotate List

标签:

原文地址:http://blog.csdn.net/kaitankedemao/article/details/44498705

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!