力扣练习——旋转链表

2 旋转链表

1.问题描述
给定一个链表，旋转链表，将链表每个节点向右移动 k 个位置，其中 k 是非负数。

示例 1:

输入: 1->2->3->4->5->NULL, k = 2

输出: 4->5->1->2->3->NULL

解释:

向右旋转 1 步: 5->1->2->3->4->NULL

向右旋转 2 步: 4->5->1->2->3->NULL

示例 2:

输入: 0->1->2->NULL, k = 4

输出: 2->0->1->NULL

解释:

向右旋转 1 步: 2->0->1->NULL

向右旋转 2 步: 1->2->0->NULL

向右旋转 3 步: 0->1->2->NULL

向右旋转 4 步: 2->0->1->NULL

可使用以下代码，完成其中的rotateRight函数，其中形参head指向无头结点单链表，k为旋转的步数，返回旋转后的链表头指针。

#include

using namespace std;

struct ListNode

{

int val;

ListNode *next;

ListNode() : val(0), next(NULL) {}

ListNode(int x) : val(x), next(NULL) {}

ListNode(int x, ListNode *next) : val(x), next(next) {}

};

class Solution {

public:

ListNode* rotateRight(ListNode* head, int k) {

        //完成本函数

}

};

ListNode *createByTail()

{

ListNode *head;

ListNode *p1,*p2;

int n=0,num;

int len;

cin>>len;

head=NULL;

while(n<len && cin>>num)

{

    p1=new ListNode(num);

    n=n+1;

    if(n==1)

        head=p1;

    else

        p2->next=p1;

    p2=p1;

}

return head;

}

void displayLink(ListNode *head)

{

ListNode *p;

p=head;

cout<<"head-->";

while(p!= NULL)

{

    cout<<p->val<<"-->";

    p=p->next;

}

cout<<"tail\n";

}

int main()

{

int k;

ListNode* head = createByTail();

cin>>k;

head=Solution().rotateRight(head,k);

displayLink(head);

return 0;

}
2.输入说明
首先输入链表长度len，然后输入len个整数，以空格分隔。

再输入非负整数k。
3.输出说明
输出格式见范例
4.范例
输入
5
1 2 3 4 5
2
输出
head–>4–>5–>1–>2–>3–>tail

5.代码

#include<iostream>
using namespace std;

struct ListNode

{

	int val;

	ListNode *next;

	ListNode() : val(0), next(NULL) {}

	ListNode(int x) : val(x), next(NULL) {}

	ListNode(int x, ListNode *next) : val(x), next(next) {}

};



class Solution {

public:

	ListNode* rotateRight(ListNode* head, int k) {


		//1.特殊情形
		if (head == NULL ) return NULL;

		//2.计算长度并设置为循环链表
		int len = 1;
		struct ListNode *p = head;
		struct ListNode *nhead = head;
		while (p->next)//注意这里为p->next，对于前面两种特殊情况的否定
		{
			len++;
			p = p->next;
		}
		p->next = head;//设置成循环链表
		//3.计算移动次数
        while (k >= len) 
		{
			k -= len;
		}
		//4.寻找链表新头结点
		for (int i = len - k; i > 1; i--) {
			head = head->next;
		}
		nhead = head->next;//新头结点
		head->next = NULL;//新尾结点
		return nhead;

		
		/*
       //3.计算移动真正长度[会超时]
		if (k%len == 0)
			return head;
		else
			k = k % len;
        
		//4.在循环链表中寻找新的链表头部和尾部
		ListNode *new_rear = head;
		for (int i = 0; i < len - k - 1; i++)
			head = head->next;
		nhead = head->next;
		head->next = NULL;
		return nhead;
*/


	}
};

ListNode *createByTail()

{

	ListNode *head;

	ListNode *p1, *p2 = NULL;

	int n = 0, num;

	int len;

	cin >> len;

	head = NULL;

	while (n<len && cin >> num)

	{

		p1 = new ListNode(num);

		n = n + 1;

		if (n == 1)

			head = p1;

		else

			p2->next = p1;

		p2 = p1;

	}

	return head;

}



void  displayLink(ListNode *head)

{

	ListNode *p;

	p = head;

	cout << "head-->";

	while (p != NULL)

	{

		cout << p->val << "-->";

		p = p->next;

	}

	cout << "tail\n";

}

int main()

{

	int k;

	ListNode* head = createByTail();

	cin >> k;

	head = Solution().rotateRight(head, k);

	displayLink(head);

	return 0;

}