旋转链表-LintCode

最新推荐文章于 2018-08-31 20:29:16 发布

原创最新推荐文章于 2018-08-31 20:29:16 发布 · 188 阅读

0 ·

CC 4.0 BY-SA版权

LintCode 专栏收录该内容

428 篇文章

订阅专栏

本文介绍了一种链表旋转算法，该算法将链表中的每个节点向右移动k个位置，其中k是非负整数。通过示例展示了如何将链表1->2->3->4->5->null经过旋转后变为4->5->1->2->3->null。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给定一个链表，旋转链表，使得每个节点向右移动k个位置，其中k是一个非负数
样例
给出链表1->2->3->4->5->null和k=2
返回4->5->1->2->3->null

#ifndef C170_H
#define C170_H
#include<iostream>
using namespace std;
struct ListNode{
    int val;
    ListNode *next;
    ListNode(int val)
    {
        this->val = val;
        this->next = NULL;
    }
};
class Solution {
public:
    /**
    * @param head: the list
    * @param k: rotate to the right k places
    * @return: the list after rotation
    */
    ListNode *rotateRight(ListNode *head, int k) {
        // write your code here
        if (k == 0)
            return head;
        ListNode *p = head;
        int len = 0;
        while (p != NULL)
        {
            len++;
            p = p->next;
        }
        if (len == 0)
            return head;
        if (len >= k)
            len -= k;
        else
        {
            k = k%len;
            if (k == 0)
                return head;
            len -= k;
        }
        if (len <= 0)
            return head;
        ListNode *q = head;;
        while (len != 1)
        {
            len--;
            q = q->next;
        }
        ListNode *node = q->next;
        ListNode *l = node;
        while (l->next != NULL)
        {
            l = l->next;
        }
        l->next = head;
        q->next = NULL;
        return node;
    }
};
#endif