6-1 链式表的按序号查找 (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 • 1 • 2 • 3
输出样例:
4 NA 1 2 5 3
题解
ElementType FindKth(List L, int K) { // K是索引值+1 设置count计数时候也要从1 开始 int count = 1; PtrToLNode p = L; while (L != NULL && count < K) { p = p->Next; count++; } if (p&&count == K) return p->Data; else return ERROR; }