PTA数据结构与算法题目集(中文) 7-19

PTA数据结构与算法题目集(中文)  7-19

7-19 求链式线性表的倒数第K项 (20 分)

 

给定一系列正整数，请设计一个尽可能高效的算法，查找倒数第K个位置上的数字。

输入格式:

输入首先给出一个正整数K，随后是若干正整数，最后以一个负整数表示结尾（该负数不算在序列内，不要处理）。

输出格式:

输出倒数第K个位置上的数据。如果这个位置不存在，输出错误信息NULL。

输入样例:

4 1 2 3 4 5 6 7 8 9 0 -1

输出样例:

7
题目分析：看到高效就想用数组来 因为数组按秩查找是相当快的 后面搜了一下别人的做法 更好

#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>
#include<malloc.h>

int Size;
int main()
{
    int k;
    int*Array = (int*)malloc(sizeof(int) * 1000000);
    scanf("%d", &k);
    while (1)
    {
        int num;
        scanf("%d", &num);
        if (num == -1)
            break;
        Array[Size++] = num;
    }
    k = Size - k;
    if (k <0||k>=Size)
        printf("NULL");
    else
        printf("%d", Array[k]);
    free(Array);
    return 0;
}

别人的做法

#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>
#include<malloc.h>


int main()
{
    int k=0;
    int Size=0;
    scanf("%d", &k);
    if (k <=0)
    {
        printf("NULL");
        return 0;
    }
    int*Array = (int*)malloc(sizeof(int) * k); //只需存k个数
    int Flag = 0;
    while (1)
    {
        int num;
        scanf("%d", &num);
        if (num == -1)   //当输入为-1时退出
            break;
        Array[Size++] = num;
        if (!Flag && Size >= k)
            Flag = 1;
        Size%=k;
    }
    if (!Flag)
        printf("NULL");
    else
        printf("%d", Array[Size]);
    free(Array);
    return 0;
}