6-4 鏈式表的按序號查找（10 分）

本題要求實現一個函數，找到並返回鏈式表的第K個元素。

函數接口定義：

ElementType FindKth( List L, int K );

其中List結構定義如下：

typedef struct LNode *PtrToLNode;
struct LNode {
    ElementType Data;
    PtrToLNode Next;
};
typedef PtrToLNode List;

L是給定單鏈表，函數FindKth要返回鏈式表的第K個元素。如果該元素不存在，則返回ERROR。

裁判測試程序樣例：

#include <stdio.h>
#include <stdlib.h>

#define ERROR -1
typedef int ElementType;
typedef struct LNode *PtrToLNode;
struct LNode {
    ElementType Data;
    PtrToLNode Next;
};
typedef PtrToLNode List;

List Read(); /* 細節在此不表 */

ElementType FindKth( List L, int K );

int main()
{
    int N, K;
    ElementType X;
    List L = Read();
    scanf("%d", &N);
    while ( N-- ) {
        scanf("%d", &K);
        X = FindKth(L, K);
        if ( X!= ERROR )
            printf("%d ", X);
        else
            printf("NA ");
    }
    return 0;
}

/* 你的代碼將被嵌在這裏 */
 

輸入樣例：

1 3 4 5 2 -1
6
3 6 1 5 4 2

輸出樣例：

4 NA 1 2 5 3

答案：

ElementType FindKth( List L, int K )
{
　　PtrToLNode pCur = L;
　　int i = 0;
　　int ret = -1;

　　if(L == NULL || K < 0)
　　{
　　　　return ERROR;
　　}


　　for(i = 0; i < K; i++)
　　{
　　　　ret = pCur->Data;
　　　　pCur = pCur->Next;
　　　　if(pCur == NULL)
　　　　{
　　　　　　if(i == K-1)
　　　　　　{
　　　　　　　　return ret;
　　　　　　}
　　　　　　return ERROR;
　　　　}
　　}

　　return ret;

}

6-4 鏈式表的按序號查找（10 分）