1. 程式人生 > >leetcode 61. 旋轉鏈表

leetcode 61. 旋轉鏈表

找到 targe clas 頭結點 listnode problems 長度 出發 fin

題目描述:

給定一個鏈表,旋轉鏈表,將鏈表每個節點向右移動 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

思路:我的想法是定義兩個點p1,p2,p1先出發,p2等k次再出發,找到旋轉後的頭結點,和尾節點
註意:上述做法要保證k小於鏈表的長度,如果大於,則對k求余即可,如果k是鏈表長度的整倍數,則直接返回頭結點。
題目鏈接:https://leetcode-cn.com/problems/rotate-list/
/**
 * 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) { if(head == NULL || k<=0 || head->next == NULL) { return head; } ListNode *p1 = head, *p2 = head, *p = NULL;
int size=1; while(p1->next !=NULL) //求鏈表長度 { p1
= p1->next; size++; }
if(k%size == 0) //是整倍數直接返回 { return head; }
k
= k%size; p1 = head; while(p1 ->next != NULL) //移動p1,p2; { p1 = p1->next; if(k <=0) { p2 = p2->next; } else { k--; } } p = p2->next; p2->next = NULL; p1->next = head; return p; } };

 

leetcode 61. 旋轉鏈表