c++ - 返回链表中最后一个元素的第 5 个元素？

Question

以上问题是我在面试中被问到的。虽然我还在等待结果，但我想知道我的做法是否正确，或者是否有更好的解决方案？

 struct Node  
 {   
  int data;  
  Node * next;    
 };  
 Node *function( struct Node *head)   
 {    
    if ( ! head)   
    {   
        std::cout <<" no linked present";  
        return 0;  
    }  

    Node  *current, *temp;  
    temp = NULL;
    current = head;  
    int counter = 0;  
    while( current != NULL)   
    {
        counter ++;  
        current = current->next;   
        if( counter >= 5)   
        {
            if( temp == NULL)  
                temp = head;   
            else
            temp = temp->next;  
        }  

    }  
     return (temp);    
   }    

如果你们有更好的最优解，请在这里发布。谢谢大家山姆

Answer 1

对于单链表，你的解决方案应该是最优的。对于双链表，你可以从后面开始。