Practise_Codes/RemoveNthNodeFromListEnd.cpp at main · srushti-learner/Practise_Codes · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
ListNode* Solution::removeNthFromEnd(ListNode* A, int B) {
    ListNode * temp = A;
    if (temp==NULL || temp->next==NULL && B==1)
    {
        return NULL;
    }
    int count=0;
    while(temp->next != NULL)
    {
        count=count+1;
        temp=temp->next;
    }
    temp=A;
    int traverse = count-B;

    if(traverse <=0)
    {
        A=A->next;
        return A;
    }
    else{
    while(traverse>0 )
    {
        temp=temp->next;
        traverse=traverse-1;
    }

    if (temp->next->next==NULL && traverse !=0 )
    {
        temp->next=NULL;
        return A;
    }
    else
    {
        temp->next=temp->next->next;
        return A;
    }}


}