我的代码在反转链表方面有什么问题?

时间:2017-02-28 17:28:58

标签: java algorithm data-structures linked-list

我正在尝试解决一个简单的leetcode问题:

反转单链表。 link

这是我的代码:

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
public class Solution {
    public ListNode reverseList(ListNode head) {
        if (head == null)
        {
            return head;
        }
        ListNode prev = null;
        while (head != null && head.next != null)
        {
            ListNode current = head;
            current.next = prev;
            prev = head; // I think the problem is with this statement
            head = head.next;
        }
        return head;
    }
}

我要做的是遍历列表的所有节点,并在每个步骤通过将前一个节点保存在ListNode变量中将当前节点链接到上一个节点。

以下是测试用例和输出,以方便您使用。

  

输入:[1,2]输出:[]预期:[2,1]

同样,我并不是在寻找替代解决方案或递归解决方案。我只是想知道我的代码有什么问题(如果适用的话还有逻辑)。

2 个答案:

答案 0 :(得分:3)

您的代码实际上只进行了一次迭代,因为它的工作原理如下:

ListNode current = head;
// You assign head.next to prev here
// which is equal to null for the first iteration
current.next = prev;
prev = head; 
// And here you go to head.next, which was set to null above
head = head.next;
// The head is null, so the while loop ends

由于您没有要求正确的解决方案,我只是提示如何修复它:在将head.next分配给prev之前,您应该将43 12/2 01:18 Amy AB int int/int int:int string string 存储在某处。

答案 1 :(得分:2)

你忘了在覆盖之前保存head.next。

当您将头部分配给ListNode当前时,您只是分配引用,而不是复制节点。因此currenthead完全同义(并且是多余的)。