Remove every k th node of the linked list in swift

Swift program for Remove every k th node of the linked list. Here problem description and explanation.

import Foundation
// Swift 4 Program for
// delete every k-th node of the linked list

// Linked list node
class LinkNode
{
	var data: Int;
	var next: LinkNode? ;
	init(_ data: Int)
	{
		self.data = data;
		self.next = nil;
	}
}
class SingleLL
{
	var head: LinkNode? ;
	init()
	{
		self.head = nil;
	}
	// Add new node at the end of linked list
	func addNode(_ value: Int)
	{
		// Create  node
		let node: LinkNode? = LinkNode(value);
		if (self.head == nil)
		{
			self.head = node;
		}
		else
		{
			var temp: LinkNode? = self.head;
			// Find last node
			while (temp!.next  != nil)
			{
				// Visit to next node
				temp = temp!.next;
			}
			// Add node at last position
			temp!.next = node;
		}
	}
	// Display all Linked List elements
	func display()
	{
		if (self.head  != nil)
		{
			var temp: LinkNode? = self.head;
			while (temp  != nil)
			{
				// Display node value
				print(temp!.data, terminator: "  ");
				// Visit to next node
				temp = temp!.next;
			}
		}
		else
		{
			print("Empty Linked list\n");
		}
	}
	func removeKh(_ kth: Int)
	{
		if (self.head == nil)
		{
			// When no element
			print("Empty Linked list");
		}
		else if (kth <= 0)
		{
			print("\nInvalid position ",kth);
		}
		else
		{
			var temp: LinkNode? = self.head;
			var hold: LinkNode? = nil;
			var checker: LinkNode? = nil;
			var count: Int = 0;
			if (kth == 1)
			{
				// When delete all element
				self.head = nil;
			}
			while (temp  != nil)
			{
				count += 1;
				if (count % kth == 0)
				{
					// When delete node found
					hold = temp;
				}
				else
				{
					// Get current node
					checker = temp;
				}
				temp = temp!.next;
				// When get free node
				if (hold  != nil)
				{
					if (checker  != nil)
					{
						// Unlink node
						checker!.next = hold!.next;
					}
					hold = nil;
				}
			}
		}
	}
	static func main(_ args: [String])
	{
		let sll: SingleLL? = SingleLL();
		// Linked list
		// 1 → 2 → 3 → 4 → 5 → 6 → 7 → NULL
		sll!.addNode(1);
		sll!.addNode(2);
		sll!.addNode(3);
		sll!.addNode(4);
		sll!.addNode(5);
		sll!.addNode(6);
		sll!.addNode(7);
		let position: Int = 2;
		print("Position :",position);
		print("Before Delete Linked List");
		sll!.display();
		sll!.removeKh(position);
		print("\n After Linked List");
		sll!.display();
	}
}
SingleLL.main([String]());

Output

Position : 2
Before Delete Linked List
1  2  3  4  5  6  7
 After Linked List
1  3  5  7


Please share your knowledge to improve code and content standard. Also submit your doubts, and test case. We improve by your feedback. We will try to resolve your query as soon as possible.

New Comment







© 2021, kalkicode.com, All rights reserved