PTA 链式表的按序号查找 (10 分)

PTA 链式表的按序号查找 (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 
#include 

#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 
结尾无空行

---

AC:

ElementType FindKth(List L, int K)
{
    int cnt = 1;
    if(L == NULL)
        return ERROR;
    while(L && cnt < K)
    {
        L = L->Next;
        cnt++;
    }
    if(L && cnt == K)
        return L->Data;
    else
        return ERROR;
}

---