本题要求实现一个函数,找到并返回链式表的第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;
}