Java/src/test/java/com/examplehub/leetcode/middle/AddTwoNumbersTest.java at master · JavaCimcoz/Java · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
package com.examplehub.leetcode.middle;

import static org.junit.jupiter.api.Assertions.assertEquals;

import org.junit.jupiter.api.Test;

class AddTwoNumbersTest {

  @Test
  void solution1() {
    ListNode l1 = new ListNode(2);
    l1.next = new ListNode(4);
    l1.next.next = new ListNode(3);
    l1.next.next.next = null;

    ListNode l2 = new ListNode(5);
    l2.next = new ListNode(6);
    l2.next.next = new ListNode(4);
    l2.next.next.next = null;

    assertEquals(l1.next.next.val, 3);
    assertEquals(l2.next.next.val, 4);

    ListNode result = AddTwoNumbers.solution1(l1, l2);

    StringBuilder builder = new StringBuilder();
    while (result != null) {
      builder.append(result.val).append("->");
      result = result.next;
    }

    builder.append("NULL");
    assertEquals("7->0->8->NULL", builder.toString());

    l1 = new ListNode(0);
    l2 = new ListNode(0);
    result = AddTwoNumbers.solution1(l1, l2);
    builder.delete(0, builder.length());
    while (result != null) {
      builder.append(result.val).append("->");
      result = result.next;
    }
    builder.append("NULL");
    assertEquals("0->NULL", builder.toString());
  }

  @Test
  void solution2() {
    ListNode l1 = new ListNode(2);
    l1.next = new ListNode(4);
    l1.next.next = new ListNode(3);
    l1.next.next.next = null;

    ListNode l2 = new ListNode(5);
    l2.next = new ListNode(6);
    l2.next.next = new ListNode(4);
    l2.next.next.next = null;

    assertEquals(l1.next.next.val, 3);
    assertEquals(l2.next.next.val, 4);

    ListNode result = AddTwoNumbers.solution2(l1, l2);

    StringBuilder builder = new StringBuilder();
    while (result != null) {
      builder.append(result.val).append("->");
      result = result.next;
    }

    builder.append("NULL");
    assertEquals("7->0->8->NULL", builder.toString());

    l1 = new ListNode(0);
    l2 = new ListNode(0);
    result = AddTwoNumbers.solution2(l1, l2);
    builder.delete(0, builder.length());
    while (result != null) {
      builder.append(result.val).append("->");
      result = result.next;
    }
    builder.append("NULL");
    assertEquals("0->NULL", builder.toString());
  }
}