diff --git a/src/ListNode.java b/src/ListNode.java index dd6028c..8293484 100644 --- a/src/ListNode.java +++ b/src/ListNode.java @@ -1,3 +1,4 @@ public class ListNode { - + public char data; + public ListNode next; } \ No newline at end of file diff --git a/src/Practice.java b/src/Practice.java index 34a2f8d..22f540c 100644 --- a/src/Practice.java +++ b/src/Practice.java @@ -1,5 +1,77 @@ public class Practice { public static void main(String[] args) { + ListNode myNode = new ListNode(); + myNode.data = 'x'; + ListNode anotherNode = new ListNode(); + anotherNode.data = 't'; + + myNode.next = anotherNode; + + ListNode moreNode = new ListNode(); + moreNode.data = 'r'; + + anotherNode.next = moreNode; + + ListNode coolNode = new ListNode(); + coolNode.data = 'w'; + coolNode.next = myNode; + + ListNode extra = moreNode; + moreNode.data = 'e'; + + ListNode hello = new ListNode(); + hello.data = 'x'; + extra.next = hello; + + System.out.println("Original list"); + printList(coolNode); + System.out.println("\nList with 't' removed"); + ListNode listWithAnotherNodeRemoved = removeAt(coolNode, 2); + printList(listWithAnotherNodeRemoved); + } + + public static void printList(ListNode head) { + ListNode current = head; + + while (current != null) { + System.out.println(current.data); + current = current.next; + } + } + + // Count how many nodes hold an x + public static int countX(ListNode head) { + ListNode current = head; + int xCount = 0; + + while (current != null) { + if (current.data == 'x') { + xCount++; + } + current = current.next; + } + + return xCount; + } + + public static ListNode removeAt(ListNode head, int removeIndex) { + int ind = 0; + ListNode current = head; + ListNode previous = null; + + while (current != null) { + if (ind == removeIndex) { + if (previous == null) { //Removing the first element? Just return the second element as the new head + return head.next; + } + previous.next = current.next; //Point the previous element to the next element, skipping this one + return head; //Head was not modified + } + previous = current; + current = current.next; + ind++; + } + return head; //Head was not modified } }