问题是我在完成LeetCode第23题时遇到的,我采用的是最小堆加上递归的解决方式。递归中,我对参数数组lists做了foreach,并且赋值,发现lists的值不会改变,结果是栈溢出。
当改成直接修改数组元素时AC。
for (int i = 0; i < lists.length; i++) { if (lists[i] != null && minVal == lists[i].val) { pre.next = lists[i]; lists[i] = lists[i].next; // 新链表头不为空,再次加入最小堆 if (lists[i] != null) { minHeap.add(lists[i].val); } break; } }
反编译两个代码,可以发现不同。foreach操作的不是原数组,而是一个拷贝数组,对数组元素的赋值相当于对一个临时变量赋值,没有影响到原数组的值。
// 普通for循环 private static void mergeKLists(D23.ListNode[] lists, PriorityQueue<Integer> minHeap, D23.ListNode pre) { if (minHeap.isEmpty()) { pre.next = null; } else { Integer minVal = (Integer)minHeap.poll(); for(int i = 0; i < lists.length; ++i) { if (lists[i] != null && minVal == lists[i].val) { pre.next = lists[i]; lists[i] = lists[i].next; if (lists[i] != null) { minHeap.add(lists[i].val); } break; } } mergeKLists(lists, minHeap, pre.next); } }
// foreach private static void mergeKLists(D23.ListNode[] lists, PriorityQueue<Integer> minHeap, D23.ListNode pre) { if (minHeap.isEmpty()) { pre.next = null; } else { Integer minVal = (Integer)minHeap.poll(); D23.ListNode[] var4 = lists; int var5 = lists.length; for(int var6 = 0; var6 < var5; ++var6) { D23.ListNode cur = var4[var6]; if (cur != null && minVal == cur.val) { pre.next = cur; cur = cur.next; if (cur != null) { minHeap.add(cur.val); } break; } } mergeKLists(lists, minHeap, pre.next); } }