Remove Nth Node From beginning of List

Write a program to Remove Nth Node From the beginning of List

Example 1:

Input: 5->2->8->4->1->NULL, n=3
Output:5->2->4->1->NULL

Approach:

Java

public class LinkListDeletionNth {
    public static void main(String[] args) {
        ListNode l = new ListNode(1);
        l.next = new ListNode(2);
        l.next.next = new ListNode(3);
        l.next.next.next = new ListNode(4);
        l.next.next.next.next = new ListNode(5);
        int n = 1;
        ListNode node = removetNthNode(l, n);
        System.out.println("After Removing :");
        linkedListTraversal(node);
    }

    private static void linkedListTraversal(ListNode node) {
        while (node != null) {
            System.out.print(node.val + " ");
            node = node.next;
        }
    }

    public static ListNode removetNthNode(ListNode head, int n) {
        // If list is null
        if (head == null || n == 0) {
            return head;
        }
        // if delete first element
        // Base case
        if (n == 1) {
            return head.next;
        }

        int count = 0;
        ListNode root = head;
        ListNode prev = head;
        while (root != null) {
            count++;
            // if found nth then delete
            if (count == n) {
                prev.next = root.next;
                break;
            }
            // change previous
            prev = root;
            // head->next
            root = root.next;
        }
        return head;
    }
}

class ListNode {

    int val;
    ListNode next;

    ListNode() {
    }

    ListNode(int val) {
        this.val = val;
    }

    ListNode(int val, ListNode next) {
        this.val = val;
        this.next = next;
    }
}

// Time Complexity : O(n)
// Space Complexity : O(1)

C++

#include <bits/stdc++.h>
using namespace std;
//Struture of  node
struct Node
{ 
   int data;
   Node *next;
   Node(int data)
    {
        this->data=data;
        this->next=NULL;
    }
};

//function to remove nth node from beginning

Node* removetNthNode(Node* head, int n) 
{
        // If list is nullptr
        if (head == NULL || n == 0) {
            return head;
        }
        // if delete first element
        // Base case
        if (n == 1) {
            return head->next;
        }

        int count = 0;
        Node *root = head;
        Node *prev = head;
        while (root != NULL) {
            count++;
            // if found nth then delete
            if (count == n) {
                prev->next = root->next;
                free(root);
                break;
            }
            // change previous
            prev = root;
            // head->next
            root = root->next;
        }
    return head;
}
//function to traverse(print) a linked 
//list
void linkedListTraversal(Node *head)
{
    Node *temp=head;
    //iterate till we reach end of the linked 
    //list
    while(temp!=NULL)
     {
         //print the current
         //node data
         cout<<temp->data<<" ";
         //move to next node
         temp=temp->next;
     }
}
int main()
{
  Node *head=new Node(5);
  head->next=new Node(2);
  head->next->next=new Node(8);
  head->next->next->next=new Node(4);
  head->next->next->next->next=new Node(1);
  int n=3;
  head=removetNthNode(head,n);
  cout<<"After Removing \n";
  linkedListTraversal(head);
  return 0;
}

// Time Complexity : O(n)
// Space Complexity : O(1)