在线编程|将链表后k个节点移到链表头

  • 问题描述
    给一个链表和整数k,将后k个节点移到链表头。
    For example:
    Given1->2->3->4->5->NULL , k =2,
    return4->5->1->2->3->NULL.
  • 解题思路
    1.将单链表变成循环链表
    先遍历一遍,得出链表节点个数count,将尾结点next指针指向首节点,形成一个环
    2.重新确定链表头和链表尾。
    接着往后跑lcount-k%count(k可能大于链表节点数)步,断开,返回结果。
  • 代码实现
class Solution {
public:
    ListNode *rotateRight(ListNode *head, int k) {
        if(NULL == head || k == 0)
            return head;

        //统计链表节点数,并将其变成循环链表
        int count = 1;
        ListNode *_head = head;
        while(NULL != _head -> next)
        {
            count++;
            _head = _head -> next;
        }
        _head -> next = head;//首尾相连

        //确定新链表头,并断开循环链表
        k = count - k % count;//k可能大于链表节点数,令k=k%count
        while(k--)
        {
            _head = _head -> next;
        }
        head = _head -> next;//新链表头
        _head -> next = NULL;//断开

        return head;
    }
};

猜你喜欢

转载自blog.csdn.net/ananbei/article/details/80627486