链式表的按序号查找

链式表的按序号查找

本题要求实现一个函数，找到并返回链式表的第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 p = L; 
    int i = 1; /* K从1开始，所以这里计数从1开始 */
    while (p != NULL && i < K) { /* 当p不为空和i<k时 */
        p = p->Next;
        i++;
    }/* 结束循环，满足两个条件之一 */
    if (i == K && p) /* 有可能i=K时，p=NULL，这时返回p->Data就会产生越界访问 */
        return p->Data;
    return ERROR;
}