一道关于单链表查找的题

给定链表的类型如下

typedef int ElementType;
typedef struct Node{
	ElementType data;
	struct Node  *next;
}Node, *LinkList;

设计Node* locate(LinkList L, int i)函数。
该函数在带头节点的单链表L中查找第i个节点。如果存在，返回该结点的指针，否则返回空指针。
请注意，本题有预置代码，只需提交所要求的函数定义代码即可。

预置代码

#include<bits/stdc++.h>
using namespace std;

typedef int ElementType;
typedef struct Node{
	ElementType data;
	struct Node  *next;
}Node, *LinkList;

Node* locate(LinkList L, int i);

int main()
{
    /*此处代码由测试程序自动添加，主要为了向顺序表中插入数据
	并输出数据,你无需关心此处代码的具体实现细节。
	如果有必要，请自己添加代码以测试你的函数是否正确。 
    */
    return 0;
}
/*你的提交的代码将被添加在此处，请完成题目所要求的函数的定义*/

这是错误答案：

Node* locate(LinkList L, int i){
    Node *p;
    if(i<1){
        return NULL;
    }
    p=locate(L,i);
    return p;
}

这是正确答案：

Node* locate(LinkList L, int i)
{
    Node *p;
    p=L;
    int k=0;
    if(i<1){
        return NULL;
    }
    while((p!=NULL) && (k<i)){
        p=p->next;
        k++;
    }
    return p;
}

第一种答案oj判定超出时间，

但我不知道错在什么地方，求解啊啊啊

> 如果有必要，请自己添加代码以测试你的函数是否正确。

你自己写代码测试过么？