6-4 链式表的按序号查找(PTA)

本题要求实现一个函数，找到并返回链式表的第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)
{
	int cnt=1;//带头节点，第一个标号就是1 
	PtrToLNode p;
	p=L;
	while(p&&cnt<K)//如果p非空或者标号<K都要继续往后走 
	{
		p=p->Next;
		cnt++;
	}
	if(p&&cnt==K)//确保非空和标号相等 
	return p->Data;
	else
	return 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;
}
List Read()
{
    PtrToLNode head,L,e;
    head=(List)malloc(sizeof(struct LNode));//申请节点空间 
    L=head=NULL;//头结点置空 
    ElementType x;
    while(1)
	{
		scanf("%d",&x); 
		if(x==-1)//-1为结束标志
		break;
        e=(List)malloc(sizeof(struct LNode));//读入的数据存放要申请新节点空间 
        e->Data=x;
        e->Next=NULL;
        if(L==NULL)//这里要区分，如果头结点为空，要给它一个非空节点 
		{
            L=head=e;
        }
		else
		{
            L->Next=e;
        }
        L=e;
    }
    return head;
}
ElementType FindKth(List L,int K)
{
	int cnt=1;//带头节点，第一个标号就是1 
	PtrToLNode p;
	p=L;
	while(p&&cnt<K)//如果p非空或者标号<K都要继续往后走 
	{
		p=p->Next;
		cnt++;
	}
	if(p&&cnt==K)//确保非空和标号相等 
	return p->Data;
	else
	return ERROR;
}