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

6-4 链式表的按序号查找(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

输出样例:

4 NA 1 2 5 3

答案:

ElementType FindKth( List L, int K )
{
  PtrToLNode pCur = L;
  int i = 0;
  int ret = -1;

  if(L == NULL || K < 0)
  {
    return ERROR;
  }


  for(i = 0; i < K; i++)
  {
    ret = pCur->Data;
    pCur = pCur->Next;
    if(pCur == NULL)
    {
      if(i == K-1)
      {
        return ret;
      }
      return ERROR;
    }
  }

  return ret;

}

 
posted @ 2018-02-04 18:02  123boy  阅读(746)  评论(0编辑  收藏  举报