删除链接列表中的所有值等于给定值

时间:2016-03-26 22:29:57

标签: java algorithm data-structures linked-list singly-linked-list

你能否澄清为什么我的代码不能总是删除链表中的所有值等于方法参数中的给定值?我该如何解决?它通过了97%的测试用例。我更喜欢修复此问题而不是使用prev / next / dummy指针更改整个方法。

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
public class Solution {
    /**
     * @param head a ListNode
     * @param val an integer
     * @return a ListNode
     */
    public ListNode removeElements(ListNode head, int val) {
        while (head!=null && head.val==val){
                head = head.next;
        }
        ListNode tmp=head;

        while (tmp!=null) {
                if (tmp.next!=null && tmp.next.val== val ) {
                    tmp.next=tmp.next.next;
                }
            tmp=tmp.next;
        }
        if (tmp != null) {
            if (tmp.val == val) {
                tmp = tmp.next;
            }
        }
        return head;
    }
}

它没有通过此测试用例:

Input
5->6->6->null, 6
Output
5->6->null
Expected
5->null

以及the problem更详细信息: Given 1->2->3->3->4->5->3, val = 3, you should return the list as 1->2->4->5

2 个答案:

答案 0 :(得分:5)

在内部while循环中,更改:

if (tmp.next!=null && tmp.next.val== val ) {
     tmp.next=tmp.next.next;
}

while (tmp.next!=null && tmp.next.val== val ) {
     tmp.next=tmp.next.next;
}

您的版本将跳过要删除的每对连续值中的第二个值。 你做了什么:

5→6-> 6->空

-tmp:5 - >先删除6,然后将tmp设置为第二个6

-tmp:6,tmp.next:null - >完成(剩下一个6)

答案 1 :(得分:0)

我认为这也有效:

while (tmp!=null) {
     if (tmp.next!=null && tmp.next.val== val ) {
         tmp.next=tmp.next.next;
      }
      else tmp=tmp.next;
 }

http://code.geeksforgeeks.org/K5yVzm