LintCode之删除链表中的元素

题目描述

我的代码

 /**
  * 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) {
         // write your code here
        if(head == null) {
            return null;
        }
        //新建一个头节点
        ListNode h = new ListNode(0);
        ListNode p = h;
        while(head != null) {
            if(head.val != val) {
                p.next = head;
                p = p.next;
            }
            if(head.next != null) {
                head = head.next;
            }else {
                break;
            }
        }

        /*
        * 这里要特别注意，当最后一个节点是要删除的节点时，
        * 没有这个判断测试数据不能通过
        */
        if(p.next != null) {
            p.next = null;
        }

        if(h.next == null) {
            return null;
        }else {
            h = h.next;
            return h;
        }
     }
 }    

结论：删除链表中的元素这一道题是LintCode中的一道入门题，以前写算法题的时候用的都是C语言写的，C语言中有指针这个概念，利用指针理解链表的结构就比较容易，但是这次我用的是Java，刚开始写的时候一脸蒙，转不过弯来。我的思路是新建一个头节点h，再把这个节点赋值给p，然后循环遍历链表，当一个节点的值不等于要删除的值时，让p的next指向这个节点，再让p=p.next，最后特别注意的就是当链表的最后一个节点的值等于要删除的节点时。