【数据结构习题】求链表的倒数第m个元素(PTA)

要求:

请设计时间和空间上都尽可能高效的算法,在不改变链表的前提下,求链式存储的线性表的倒数第m(>0)个元素。

函数接口定义:

ElementType Find( List L, int m );

其中List结构定义如下:

typedef struct Node *PtrToNode;
struct Node {
    ElementType Data; /* 存储结点数据 */
    PtrToNode   Next; /* 指向下一个结点的指针 */
};
typedef PtrToNode List; /* 定义单链表类型 */

L是给定的带头结点的单链表;函数Find要将L的倒数第m个元素返回,并不改变原链表。如果这样的元素不存在,则返回一个错误标志ERROR

裁判测试程序样例:

#include 
#include 

#define ERROR -1

typedef int ElementType;
typedef struct Node *PtrToNode;
struct Node {
    ElementType Data;
    PtrToNode   Next;
};
typedef PtrToNode List;

List Read(); /* 细节在此不表 */
void Print( List L ); /* 细节在此不表 */

ElementType Find( List L, int m );

int main()
{
    List L;
    int m;
    L = Read();
    scanf("%d", &m);
    printf("%d\n", Find(L,m));
    Print(L);
    return 0;
}

/* 你的代码将被嵌在这里 */

输入样例:

5
1 2 4 5 6
3

输出样例:

4
1 2 4 5 6 

完整代码

#include 
#include 

#define ERROR -1

typedef int ElementType;
typedef struct Node *PtrToNode;
struct Node {
    ElementType Data;
    PtrToNode   Next;
};
typedef PtrToNode List;

List Read(); /* 细节在此不表 */
void Print( List L ); /* 细节在此不表 */

ElementType Find( List L, int m );

int main()
{
    List L;
    int m;
    L = Read();
    scanf("%d", &m);
    printf("%d\n", Find(L,m));
    Print(L);
    return 0;
}

/* 你的代码将被嵌在这里 */

/*
 * @Description : 创建一个用户定义的链表
 * @param       : 无
 * @return 		: 新的的链表
 */
List Read()
{
    int n;
    scanf("%d", &n);
    //为头节点申请空间
    List Head = (List)malloc(sizeof(PtrToNode));
    //防止野指针的出现
    Head->Next = NULL;
    //当n不是0的时候
    if (n != 0){
        //创建一个指针
        List p = Head;
        for (int i = 0; i < n; i++){
            //创建一个中间节点
            List Var = (List)malloc(sizeof(PtrToNode));
            //尾插法
            scanf("%d", &(Var->Data));
            //p的指针指向新的元素
            p->Next = Var;
            //p后移动
            p = Var;
        }
        //安全起见,尾指针为空
        p->Next = NULL;
    }
    return Head;
}

/*
 * @Description : 打印链表
 * @param  -L   : 需要打印的链表
 * @return 		: 无
 */
void Print( List L )
{
    //先找到第一个结点,判断是不是空
    L = L->Next;
    if (L == NULL){
        printf("NULL");
    }
    else{
        while (L){
            printf("%d ",L->Data);
            L = L->Next;
        }
    }
    printf("\n");
}

/*
 * @Description : 查找函数
 * @param  -L   : 需要查找的链表
 * @param  -m   : 第m个元素
 * @return 		: 返回找到的元素
 *                不存在返回错误
 */
ElementType Find( List L, int m )
{
    int Len = 0;
    int i = 0;
    List p = L;
    //计算链表长度
    while (p->Next){
        Len++;
        p = p->Next;
    }
    if (m > Len){
        return ERROR;
    }
    else{
        p = L;
        while (p->Next && i < Len - m ){
            p = p->Next;
            i++;
        }
        return p->Next->Data;
    }
}


最后

该方法不是最优解,耗时26ms,如果大佬有更好的方法,请多多指教!

你可能感兴趣的:(C++,数据结构,链表,算法)