package test.leecode.linkedlist; import org.junit.Assert; import org.junit.Test; import cn.fansunion.leecode.linkedlist.ListNode; import cn.fansunion.leecode.linkedlist.RemoveLinkedListElements; /** * @author wen.lei@brgroup.com * * 2022-2-23 */ public class RemoveLinkedListElementsTest { /* 输入:head = [1,2,6,3,4,5,6], val = 6 输出:[1,2,3,4,5] 示例 2: 输入:head = [], val = 1 输出:[] 示例 3: 输入:head = [7,7,7,7], val = 7 输出:[]*/ @Test public void test1263456() { RemoveLinkedListElements list = new RemoveLinkedListElements(); //1,1,1,2,3,3 ListNode node62 = new ListNode( 6 , null ); ListNode node5 = new ListNode( 5 ,node62); ListNode node4 = new ListNode( 4 ,node5); ListNode node3 = new ListNode( 3 ,node4); ListNode node61 = new ListNode( 6 ,node3); ListNode node2 = new ListNode( 2 ,node61); ListNode head1 = new ListNode( 1 ,node2); ListNode newHead=list.removeElements(head1, 6 ); Assert.assertEquals( 1 , newHead.val); Assert.assertEquals( 2 , newHead.next.val); Assert.assertEquals( 3 , newHead.next.next.val); Assert.assertEquals( 4 , newHead.next.next.next.val); Assert.assertEquals( 5 , newHead.next.next.next.next.val); } @Test public void test7777() { RemoveLinkedListElements list = new RemoveLinkedListElements(); //7,7,7,7 ListNode node73 = new ListNode( 7 , null ); ListNode node72 = new ListNode( 7 ,node73); ListNode node71 = new ListNode( 7 ,node72); ListNode head7 = new ListNode( 7 ,node71); ListNode newHead=list.removeElements(head7, 7 ); Assert.assertNull(newHead); } @Test public void test2777() { RemoveLinkedListElements list = new RemoveLinkedListElements(); //7,7,7,7 ListNode node73 = new ListNode( 7 , null ); ListNode node72 = new ListNode( 7 ,node73); ListNode node71 = new ListNode( 7 ,node72); ListNode head2 = new ListNode( 2 ,node71); ListNode newHead=list.removeElements(head2, 7 ); Assert.assertEquals( 2 , newHead.val); Assert.assertNull(newHead.next); } } |