leetcode 61. 旋轉鏈表
阿新 • • 發佈:2019-05-02
找到 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. 旋轉鏈表