Delete N nodes after M nodes of a linked list

Given a linked list and two integers M and N. Traverse the linked list such that you retain M nodes then delete next N nodes, continue the same till end of the linked list.

Difficulty Level: Rookie

Examples:

Input:
M = 2, N = 2
Linked List: 1->2->3->4->5->6->7->8
Output:
Linked List: 1->2->5->6

Input:
M = 3, N = 2
Linked List: 1->2->3->4->5->6->7->8->9->10
Output:
Linked List: 1->2->3->6->7->8

Input:
M = 1, N = 1
Linked List: 1->2->3->4->5->6->7->8->9->10
Output:
Linked List: 1->3->5->7->9

Recommended: Please solve it on “PRACTICE” first, before moving on to the solution.

The main part of the problem is to maintain proper links between nodes, make sure that all corner cases are handled. Following is C implementation of function skipMdeleteN() that skips M nodes and delete N nodes till end of list. It is assumed that M cannot be 0.

C++

// C++ program to delete N nodes 
// after M nodes of a linked list  
#include <bits/stdc++.h> 
using namespace std; 
  
// A linked list node  
class Node  
{  
    public: 
    int data;  
    Node *next;  
};  
  
/* Function to insert a node at the beginning */
void push(Node ** head_ref, int new_data)  
{  
    /* allocate node */
    Node* new_node = new Node(); 
  
    /* put in the data */
    new_node->data = new_data;  
  
    /* link the old list off the new node */
    new_node->next = (*head_ref);  
  
    /* move the head to point to the new node */
    (*head_ref) = new_node;  
}  
  
/* Function to print linked list */
void printList(Node *head)  
{  
    Node *temp = head;  
    while (temp != NULL)  
    {  
        cout<<temp->data<<" ";  
        temp = temp->next;  
    }  
    cout<<endl;  
}  
  
// Function to skip M nodes and then 
// delete N nodes of the linked list.  
void skipMdeleteN(Node *head, int M, int N)  
{  
    Node *curr = head, *t;  
    int count;  
  
    // The main loop that traverses 
    // through the whole list  
    while (curr)  
    {  
        // Skip M nodes  
        for (count = 1; count < M &&  
                curr!= NULL; count++)  
            curr = curr->next;  
  
        // If we reached end of list, then return  
        if (curr == NULL)  
            return;  
  
        // Start from next node and delete N nodes  
        t = curr->next;  
        for (count = 1; count<=N && t!= NULL; count++)  
        {  
            Node *temp = t;  
            t = t->next;  
            free(temp);  
        }  
          
        // Link the previous list with remaining nodes  
        curr->next = t;  
  
        // Set current pointer for next iteration  
        curr = t;  
    }  
}  
  
// Driver code  
int main()  
{  
    /* Create following linked list  
    1->2->3->4->5->6->7->8->9->10 */
    Node* head = NULL;  
    int M=2, N=3;  
    push(&head, 10);  
    push(&head, 9);  
    push(&head, 8);  
    push(&head, 7);  
    push(&head, 6);  
    push(&head, 5);  
    push(&head, 4);  
    push(&head, 3);  
    push(&head, 2);  
    push(&head, 1);  
  
    cout << "M = " << M<< " N = " << N << "\nGiven Linked list is :\n";  
    printList(head);  
  
    skipMdeleteN(head, M, N);  
  
    cout<<"\nLinked list after deletion is :\n";  
    printList(head);  
  
    return 0;  
}  
  
// This code is contributed by rathbhupendra 

C

// C program to delete N nodes after M nodes of a linked list 
#include <stdio.h> 
#include <stdlib.h> 
  
// A linked list node 
struct Node 
{ 
    int data; 
    struct Node *next; 
}; 
  
/* Function to insert a node at the beginning */
void push(struct Node ** head_ref, int new_data) 
{ 
    /* allocate node */
    struct Node* new_node = (struct Node*) malloc(sizeof(struct Node)); 
  
    /* put in the data  */
    new_node->data  = new_data; 
  
    /* link the old list off the new node */
    new_node->next = (*head_ref); 
  
    /* move the head to point to the new node */
    (*head_ref)  = new_node; 
} 
  
/* Function to print linked list */
void printList(struct Node *head) 
{ 
    struct Node *temp = head; 
    while (temp != NULL) 
    { 
        printf("%d ", temp->data); 
        temp = temp->next; 
    } 
    printf("\n"); 
} 
  
// Function to skip M nodes and then delete N nodes of the linked list. 
void skipMdeleteN(struct Node  *head, int M, int N) 
{ 
    struct Node *curr = head, *t; 
    int count; 
  
    // The main loop that traverses through the whole list 
    while (curr) 
    { 
        // Skip M nodes 
        for (count = 1; count<M && curr!= NULL; count++) 
            curr = curr->next; 
  
        // If we reached end of list, then return 
        if (curr == NULL) 
            return; 
  
        // Start from next node and delete N nodes 
        t = curr->next; 
        for (count = 1; count<=N && t!= NULL; count++) 
        { 
            struct Node *temp = t; 
            t = t->next; 
            free(temp); 
        } 
        curr->next = t; // Link the previous list with remaining nodes 
  
        // Set current pointer for next iteration 
        curr = t; 
    } 
} 
  
// Driver program to test above functions 
int main() 
{ 
    /* Create following linked list 
      1->2->3->4->5->6->7->8->9->10 */
    struct Node* head = NULL; 
    int M=2, N=3; 
    push(&head, 10); 
    push(&head, 9); 
    push(&head, 8); 
    push(&head, 7); 
    push(&head, 6); 
    push(&head, 5); 
    push(&head, 4); 
    push(&head, 3); 
    push(&head, 2); 
    push(&head, 1); 
  
    printf("M = %d, N = %d \nGiven Linked list is :\n", M, N); 
    printList(head); 
  
    skipMdeleteN(head, M, N); 
  
    printf("\nLinked list after deletion is :\n"); 
    printList(head); 
  
    return 0; 
} 

Java

// Java program to delete N nodes  
// after M nodes of a linked list  
import java.util.*; 
  
class GFG 
{ 
  
// A linked list node  
static class Node  
{  
    int data;  
    Node next;  
};  
  
/* Function to insert a node at the beginning */
static Node push( Node head_ref, int new_data)  
{  
    /* allocate node */
    Node new_node = new Node();  
  
    /* put in the data */
    new_node.data = new_data;  
  
    /* link the old list off the new node */
    new_node.next = (head_ref);  
  
    /* move the head to point to the new node */
    (head_ref) = new_node; 
      
    return head_ref; 
}  
  
/* Function to print linked list */
static void printList( Node head)  
{  
    Node temp = head;  
    while (temp != null)  
    {  
        System.out.printf("%d ", temp.data);  
        temp = temp.next;  
    }  
    System.out.printf("\n");  
}  
  
// Function to skip M nodes and then 
// delete N nodes of the linked list.  
static void skipMdeleteN( Node head, int M, int N)  
{  
    Node curr = head, t;  
    int count;  
  
    // The main loop that traverses 
    // through the whole list  
    while (curr!=null)  
    {  
        // Skip M nodes  
        for (count = 1; count < M && curr != null; count++)  
            curr = curr.next;  
  
        // If we reached end of list, then return  
        if (curr == null)  
            return;  
  
        // Start from next node and delete N nodes  
        t = curr.next;  
        for (count = 1; count <= N && t != null; count++)  
        {  
            Node temp = t;  
            t = t.next;  
        }  
          
        // Link the previous list with remaining nodes  
        curr.next = t;  
  
        // Set current pointer for next iteration  
        curr = t;  
    }  
}  
  
// Driver code  
public static void main(String args[]) 
{  
    /* Create following linked list  
    1.2.3.4.5.6.7.8.9.10 */
    Node head = null;  
    int M=2, N=3;  
    head=push(head, 10);  
    head=push(head, 9);  
    head=push(head, 8);  
    head=push(head, 7);  
    head=push(head, 6);  
    head=push(head, 5);  
    head=push(head, 4);  
    head=push(head, 3);  
    head=push(head, 2);  
    head=push(head, 1);  
  
    System.out.printf("M = %d, N = %d \nGiven" +  
                        "Linked list is :\n", M, N);  
    printList(head);  
  
    skipMdeleteN(head, M, N);  
  
    System.out.printf("\nLinked list after deletion is :\n");  
    printList(head);  
}  
} 
  
// This code is contributed by Arnab Kundu 

Python

# Python program to delete M nodes after N nodes 
  
# Node class  
class Node: 
  
    # Constructor to initialize the node object 
    def __init__(self, data): 
        self.data = data 
        self.next = None
  
class LinkedList: 
  
    # Function to initialize head 
    def __init__(self): 
        self.head = None
  
    # Function to insert a new node at the beginning 
    def push(self, new_data): 
        new_node = Node(new_data) 
        new_node.next = self.head 
        self.head = new_node 
  
    # Utility function to prit the linked LinkedList 
    def printList(self): 
        temp = self.head 
        while(temp): 
            print temp.data, 
            temp = temp.next
  
    def skipMdeleteN(self, M, N): 
        curr = self.head 
          
        # The main loop that traverses through the 
        # whole list 
        while(curr): 
            # Skip M nodes 
            for count in range(1, M): 
                if curr is None: 
                    return 
                curr = curr.next
                      
            if curr is None : 
                return 
  
            # Start from next node and delete N nodes 
            t = curr.next 
            for count in range(1, N+1): 
                if t is None: 
                    break
                t = t.next
      
            # Link the previous list with reamining nodes 
            curr.next = t 
            # Set Current pointer for next iteration 
            curr = t  
  
# Driver program to test above function 
  
# Create following linked list 
# 1->2->3->4->5->6->7->8->9->10 
llist = LinkedList() 
M = 2 
N = 3
llist.push(10) 
llist.push(9) 
llist.push(8) 
llist.push(7) 
llist.push(6) 
llist.push(5) 
llist.push(4) 
llist.push(3) 
llist.push(2) 
llist.push(1) 
  
print "M = %d, N = %d\nGiven Linked List is:" %(M, N) 
llist.printList() 
print 
  
llist.skipMdeleteN(M, N) 
  
print "\nLinked list after deletion is"
llist.printList() 
  
# This code is contributed by Nikhil Kumar Singh(nickzuck_007) 

C#

// C# program to delete N nodes  
// after M nodes of a linked list  
using System; 
  
class GFG  
{  
  
// A linked list node  
public class Node  
{  
    public int data;  
    public Node next;  
};  
  
/* Function to insert a node at the beginning */
static Node push( Node head_ref, int new_data)  
{  
    /* allocate node */
    Node new_node = new Node();  
  
    /* put in the data */
    new_node.data = new_data;  
  
    /* link the old list off the new node */
    new_node.next = (head_ref);  
  
    /* move the head to point to the new node */
    (head_ref) = new_node;  
      
    return head_ref;  
}  
  
/* Function to print linked list */
static void printList( Node head)  
{  
    Node temp = head;  
    while (temp != null)  
    {  
        Console.Write("{0} ", temp.data);  
        temp = temp.next;  
    }  
    Console.Write("\n");  
}  
  
// Function to skip M nodes and then  
// delete N nodes of the linked list.  
static void skipMdeleteN( Node head, int M, int N)  
{  
    Node curr = head, t;  
    int count;  
  
    // The main loop that traverses  
    // through the whole list  
    while (curr!=null)  
    {  
        // Skip M nodes  
        for (count = 1; count < M &&  
                curr != null; count++)  
            curr = curr.next;  
  
        // If we reached end of list, then return  
        if (curr == null)  
            return;  
  
        // Start from next node and delete N nodes  
        t = curr.next;  
        for (count = 1; count <= N && t != null; count++)  
        {  
            Node temp = t;  
            t = t.next;  
        }  
          
        // Link the previous list with remaining nodes  
        curr.next = t;  
  
        // Set current pointer for next iteration  
        curr = t;  
    }  
}  
  
// Driver code  
public static void Main(String []args)  
{  
    /* Create following linked list  
    1.2.3.4.5.6.7.8.9.10 */
    Node head = null;  
    int M=2, N=3;  
    head=push(head, 10);  
    head=push(head, 9);  
    head=push(head, 8);  
    head=push(head, 7);  
    head=push(head, 6);  
    head=push(head, 5);  
    head=push(head, 4);  
    head=push(head, 3);  
    head=push(head, 2);  
    head=push(head, 1);  
  
    Console.Write("M = {0}, N = {1} \nGiven" +  
                        "Linked list is :\n", M, N);  
    printList(head);  
  
    skipMdeleteN(head, M, N);  
  
    Console.Write("\nLinked list after deletion is :\n");  
    printList(head);  
}  
}  
  
// This code contributed by Rajput-Ji 

Output:

M = 2, N = 3
Given Linked list is :
1 2 3 4 5 6 7 8 9 10

Linked list after deletion is :
1 2 6 7

Time Complexity: O(n) where n is number of nodes in linked list.

This article is contributed by Chandra Prakash. Please write comments if you find anything incorrect, or you want to share more information about the topic discussed above

Attention reader! Don’t stop learning now. Get hold of all the important DSA concepts with the DSA Self Paced Course at a student-friendly price and become industry ready.

My Personal Notes

Recommended: Please solve it on “PRACTICE” first, before moving on to the solution.

C++

C

Java

Python

C#

Recommended Posts: