Added RemoveLast function and its unit test

1 files changed, 42 insertions, 0 deletions

diff --git a/CST 126/Homework_3/SinglyLinkedList.hpp b/CST 126/Homework_3/SinglyLinkedList.hpp
index 38c0072..1b47d99 100644
--- a/CST 126/Homework_3/SinglyLinkedList.hpp
+++ b/CST 126/Homework_3/SinglyLinkedList.hpp
@@ -58,4 +58,46 @@ bool Prepend(SinglyLinkedList* list, ListNode* node) {
 	return true;
 }
 
+bool RemoveFirst(SinglyLinkedList* list) {
+	if (list->_size == 0)
+		return false;
+
+	ListNode* first = list->_head;
+
+	list->_head = list->_head->_next;
+
+	list->_size--;
+
+	delete first;
+
+	return true;
+}
+
+bool RemoveLast(SinglyLinkedList* list) {
+	if (list->_size == 0)
+		return false;
+
+	/*if (list->_size == 1)
+	{
+		delete list->_head;
+		list->_head = nullptr;
+		list->_size--;
+		return true;
+	}
+	*/
+	
+	
+	ListNode* SecondToLast = list->_head;
+	while (SecondToLast->_next->_next != nullptr)
+	{
+		SecondToLast = SecondToLast->_next;
+	}
+
+	delete SecondToLast->_next;
+	SecondToLast->_next = nullptr;
+	list->_size--;
+
+	return true;
+}
+
 #endif