NC50 链表中的节点每k个一组翻转-Java

上链接这里
将给出的链表中的节点每\ k k 个一组翻转，返回翻转后的链表
如果链表中的节点数不是\ k k 的倍数，将最后剩下的节点保持原样
你不能更改节点中的值，只能更改节点本身。
要求空间复杂度 \ O(1) O(1)
例如：
给定的链表是1→2→3→4→5
对于 k = 2 你应该返回 2→1→4→3→5
对于 k=3, 你应该返回3→2→1→4→5

主要思路是用栈

需要注意的点是最后一步，需要将res.next置空，不然将会无限循环。

import java.util.*;

/*
 * public class ListNode {
 *   int val;
 *   ListNode next = null;
 * }
 */

public class Solution {
    /**
     * 
     * @param head ListNode类 
     * @param k int整型 
     * @return ListNode类
     */
    public ListNode reverseKGroup (ListNode head, int k) {
        // write code here
        ListNode res = new ListNode(0);
        ListNode resHead = res;
        ListNode cur = head;
        Stack<ListNode> s = new Stack<ListNode>();
        while(cur != null){
            int i = 0;
            // 这里要记录一下cur节点的位置
            ListNode temp = cur;
            
            while(i < k && cur != null){
                i++;
                s.push(cur);
                cur = cur.next;
            }
			//最后一轮时节点个数小于k，直接把剩余节点的头结点赋值给res即可。
            if(i < k){
                res.next = temp;
                break;
            }
            while(!s.isEmpty()){
                res.next = s.pop();
                res = res.next;
            }
            // 这个必须有，因为 3-》4翻转后变成了4-》3，但3的next还是4，就成了无限循环，会爆超时错误。
            res.next = null;
        }
        return resHead.next;
    }
}

package JZOffer;


import java.util.*;


public class Solution3 {

    public static void main(String[] args){
        ListNode head = new ListNode(1);
        head.next = new ListNode(2);
        head.next.next = new ListNode(3);
        head.next.next.next = new ListNode(4);
        int k = 2;
        ListNode res = reverseKGroup(head,k);
        while(res != null){
            System.out.println(res.val);
            res = res.next;
        }
    }

    public static ListNode reverseKGroup (ListNode head, int k) {
        // write code here
        ListNode res = new ListNode(0);
        ListNode resHead = res;
        ListNode cur = head;
        Stack<ListNode> s = new Stack<ListNode>();
        while(cur != null){
            int i = 0;
            ListNode temp = cur;

            while(i < k && cur != null){
                i++;
                s.push(cur);
                cur = cur.next;
            }

            if(i < k){
                res.next = temp;
                break;
            }
            while(!s.isEmpty()){
                res.next = s.pop();
                res = res.next;
            }
            res.next = null;
        }
        return resHead.next;
    }
}