每天一道算法题之K个一组翻转链表

K个一组翻转链表

题目描述:

给你一个链表，每 k 个节点一组进行翻转，请你返回翻转后的链表。k 是一个正整数，它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍，那么请将最后剩余的节点保持原有顺序。¹

• 例如：

  给你这个链表：1->2->3->4->5
  当 k = 2 时，应当返回: 2->1->4->3->5

  当 k = 3 时，应当返回: 3->2->1->4->5

  若数目不够无需翻转

• 思路:

  利用栈进行存储需要翻转的数据,达到K个时,再依次出栈
  对于节点个数不足K个时,进行单独处理.
  -代码部分:

public ListNode reverseKGroup(ListNode head, int k) {
    
    //本题利用栈进行求解
    if(head == null){
        return head;
    }
    
    ListNode dummy = new ListNode(0);
    Stack<ListNode> stack = new Stack<ListNode>();
    dummy.next = head;
    
    //cur 表示翻转后的最后一个元素,next表示需要翻转的第一个元素
    //链表的双指针
    ListNode cur = dummy, n = dummy.next;
    
    while(n != null){

		//存储K个即将需要翻转的节点
        int count = 0;
        while(count < k){
            stack.push(n);
            n = n.next;
            count ++;
            //当个数不足k个时
            if(n == null) break;
        }
        
        //当栈中个数小于k时不需要反转
        if(stack.size() != k) return dummy.next;
        
        //翻转
        while(stack.size() != 0){
            cur.next = stack.pop();
            cur = cur.next;
        }
        
        cur.next = n;//将翻转后的节点与下一步需要翻转的节点连接

    }

    return dummy.next;

}

---

1. https://leetcode-cn.com/problems/reverse-nodes-in-k-group/ ↩︎