链表相关算法(翻转,k个翻转)
之前在leetcode上刷题的时候碰到链表的题目就会有一点绕,这里整理一下。
链表结构
首先统一一下链表的结构
struct ListNode {
int val;
struct ListNode *next;
};
反转链表
思路
定义三个Node,分别是pre,cur,next,用于保存上一个Node,本Node以及下一个Node, 这里的pre实际上就是我们的反转链表的头节点。想法就是遍历链表,每一次都把当前的Node插到pre的前面,这样就实现了逆序,pre的初始值为nullptr
,cur的初始值是链表的头节点,第一次操作我们将头节点插入到pre的头部,即cur就指向了nullptr
,成为了最后一个Node,然后cur指向下一个节点,pre向前移动一位指向原先的cur。
代码
ListNode* reverseList(ListNode* head){
ListNode* cur = head;
ListNode* pre = nullptr;
ListNode* next = nullptr;
while(cur!=nullptr){
next = cur->next;
cur->next = pre;
pre = cur;
cur = next;
}
return pre;
}
K个反转(不足K不反转)
思路
基本和完全反转类似,先遍历一遍数组,若长度小于K,直接返回head;第二次循环执行反转操作,条件增加一个i小于k,当遍历完第k个Node时停止循环,此时cur指向了k+1个Node,而前面的K个Node已经完成反转,此时将cur作为head,递归调用reverseList函数,就可以对后面的K个Node做反转操作。
代码
ListNode* reverseKList(ListNode* head, int k){
if(head == nullptr || k <= 1) return head;
ListNode* cur=head;
ListNode* pre=nullptr;
ListNode* next=nullptr;
for(int i=0;i<k;i++){
if(cur == nullptr)
return head;
cur = cur->next;
}
cur = head;
for(int i = 0;i<k && cur != nullptr;i++){
next = cur->next;
cur->next = pre;
pre = cur;
cur = next;
}
head->next = reverseKList(cur, k);
return pre;
}
K个反转(不足K也反转)
思路
同不足K不反转,省去第一次遍历。
代码
ListNode* reverseKList(ListNode* head, int k){
if(head == nullptr || k <= 1) return head;
ListNode* cur=head;
ListNode* pre=nullptr;
ListNode* next=nullptr;
for(int i = 0;i<k && cur != nullptr;i++){
next = cur->next;
cur->next = pre;
pre = cur;
cur = next;
}
head->next = reverseKGroup(next, k);
return pre;
}