记录牛客网面试必刷TOP101—3.【BM3 】链表中的节点每k个一组翻转

/**
 * struct ListNode {
 *	int val;
 *	struct ListNode *next;
 * };
 */

class Solution {
public:
    /**
     * 
     * @param head ListNode类 
     * @param k int整型 
     * @return ListNode类
     */
    ListNode* reverseKGroup(ListNode* head, int k) {
        //找到每次翻转的尾结点
        ListNode *tail = head;
        //遍历k次到达尾结点
        for(int i=0;i<k;i++){
            //如果不足k到了链表尾，直接返回，不翻转
            if(tail==NULL){
                return head;
            }
            // 子链表尾的下一个节点
            tail = tail->next;
        }
        // 翻转时需要的前序节点和当前节点
        ListNode *pre = NULL;
        ListNode *cur = head;
        while(cur != tail){
            //翻转
            ListNode *temp = cur->next;
            cur->next = pre;
            pre = cur;
            cur = temp;
        }
        // 当前尾结点 指向下一段要翻转的链表的翻转后的头结点
        head ->next = reverseKGroup(tail,k);
        return pre;
    }
};

2

/**
 * struct ListNode {
 *	int val;
 *	struct ListNode *next;
 * };
 */
#include<stack>
class Solution {
public:
    /**
     * 
     * @param head ListNode类 
     * @param k int整型 
     * @return ListNode类
     */
    ListNode* reverseKGroup(ListNode* head, int k) {
        // 定义一个栈用于存放每k个链表元素
        stack<ListNode*> s;
        // 定义一个新链表，用于存放翻转后的链表
        ListNode *dummy = new ListNode(-1);
        ListNode *p = dummy;

        while(true){
            int count = 0;
            ListNode *temp = head;
            // 入栈
            while(temp!=NULL && count<k){
                s.push(temp);
                temp = temp->next;
                count++;
            }
            // 不足k个，直接返回剩余节点的头节点
            if(count!=k){
                p->next = head;
                break; // 循环退出条件
            }
            // 出栈
            while(!s.empty()){
                p->next = s.top();//返回栈顶元素
                s.pop();// 弹出栈顶元素
                p = p->next;
            }
            // p->next = temp;// 更新p节点，将p节点指向下一组的开始节点
            head = temp;
        }
        return dummy->next;
    }
};