java 在O(1)时间删除链表结点(剑指offer)

用O(1)的时间复杂度,删除结点

public class DeleteNodeInOne {

    public static void main(String[] args) {
        ListNode node1=new ListNode(1);
        ListNode node2=new ListNode(2);
        ListNode node3=new ListNode(3);
        ListNode node4=new ListNode(4);
        ListNode node5=new ListNode(5);
        ListNode node6=new ListNode(6);
        node1.next=node2;
        node2.next=node3;
        node3.next=node4;
        node4.next=node5;
        node5.next=node6;
        DeleteNodeInOne deleteNodeInOne=new DeleteNodeInOne();
        deleteNodeInOne.deleteNode(node1,node2);
        deleteNodeInOne.deleteNode(node1,node5);
        deleteNodeInOne.deleteNode(node1,node6);
    }

    public void deleteNode(ListNode head,ListNode node){
        if(head==null||node==null)
            return;

        //要删除的结点不是尾结点
        if(node.next!=null){
            ListNode next=node.next;
            node.val=next.val;
            node.next=next.next;
        }else if(head==node){//链表中只有一个结点,且删除尾结点
            head=null;
        }else{//链表中有多个结点,且删除尾结点
            ListNode pre=head;
            while (pre.next!=node){
                pre=pre.next;
            }
            pre.next=null;
        }
    }
}

猜你喜欢

转载自blog.csdn.net/csdnlijingran/article/details/83787896