In the given linked list find nth node. The function should return the data value in the nth node. n is input integer index.
Example
Time complexity : O(n)
Algorithm
1. Initialize count equal to 0.
2. Traverse in the linked list.
a. If count is equal to the input index then return current node data.
b. Increment count.
c. Move forward position. (continue traversing)
C++ Program
#include <bits/stdc++.h> using namespace std; struct LLNode { int data; struct LLNode* next; }; /* Function to insertAtBeginning a node */ void insertAtBeginning(struct LLNode** head, int dataToBeInserted) { struct LLNode* curr = new LLNode; curr->data = dataToBeInserted; curr->next = NULL; if(*head == NULL) *head=curr; //If this is first node make this as head of list else { curr->next=*head; //else make the curr (new) node's next point to head and make this new node a the head *head=curr; } //O(1) constant time } //display linked list void display(struct LLNode**node) { struct LLNode *temp= *node; while(temp!=NULL) { if(temp->next!=NULL) cout<<temp->data<<"->"; else cout<<temp->data; temp=temp->next; //move to next node } //O(number of nodes) cout<<endl; } int PrintNth(struct LLNode* head, int index) { struct LLNode* current = head; int count = 0; //Start travesing from head. while (current != NULL) { if(count == index) { return (current->data); } count++; current = current->next; } return 0; } //Main function int main() { struct LLNode* head = NULL;//Initial list has no elements insertAtBeginning(&head, 9); insertAtBeginning(&head, 8); insertAtBeginning(&head, 7); insertAtBeginning(&head, 6); insertAtBeginning(&head, 5); insertAtBeginning(&head, 4); insertAtBeginning(&head, 3); insertAtBeginning(&head, 2); insertAtBeginning(&head, 1); cout<<"Input linked list is: "; display(&head); int k; cout<<"Enter index: "; cin>>k; cout<<"Node at index "<< k <<" is: "<<PrintNth(head,k); return 0; }