LeetCode 61. Rotate List

AnICoo1

于 2017-06-08 19:48:21 发布

阅读量281

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签： leetcode 链表

本文链接：https://blog.csdn.net/helloiamclh/article/details/72934265

LeetCode 专栏收录该内容

74 篇文章

订阅专栏

题意

将一个链表向右旋转 $k$ 个位置

思路

直接定位旋转后的尾结点，然后将原始尾结点的 $next$ 指向原始头结点，然后将旋转后尾结点的原始 $next$ 指向的结点置为头结点.

需要注意的case:

k有可能会大于链表长度

代码

/**
 * 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) {
        ListNode *p = head;
        int cnt = 0;
        while(p != NULL){
            p = p->next;
            cnt++;
        }
        if(cnt == 0) return head;
        k %= cnt;
        if(k == 0) return head;
        cnt -= k;
        p = head;
        while(cnt > 1){
            p = p->next;
            cnt--;
        }
        ListNode *ret = p;
        ListNode *newHead = p->next;
        while(ret->next != NULL){
            ret = ret->next;
        }
        p->next = NULL;
        ret->next = head;
        return newHead;
    }
};