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

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

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

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

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

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

输出样例:

7

思路
简单来说就是他要求倒数第K个元素，我就保存K个元素，每次读入一个数据，重置一下这个数组就好了，输出的时候输出数组的第一个元素

点击访问 PTA-测验

#include<stdlib.h>
#include<stdio.h>
/* 评测结果 
时间  结果  得分  题目  编译器     用时（ms）  内存（MB）  用户
2016-08-29 10:23    答案正确    20  5-19    gcc     266     4   569985011
测试点结果 
测试点     结果  得分/满分   用时（ms）  内存（MB）
测试点1    答案正确    10/10   1   1
测试点2    答案正确    1/1     2   1
测试点3    答案正确    3/3     19  1
测试点4    答案正确    3/3     266     1
测试点5    答案正确    3/3     241     4
查看代码

int main() {
    int n;
    scanf("%d",&n);
//  printf("%d]",n);
    int* a=(int*)malloc(sizeof(int)*n);
    int count=0,flag,found=0;
    while(1) {
        scanf("%d",&flag);
        if(flag>=0) {
            a[count++]=flag;        
            if(!found){
                if(count>=n){
                found=1;count%=n;   
                }
            }else count%=n; 
        } else {
            break;
        }
    }
    if(0==found)printf("NULL");
    else printf("%d",a[count]);
    return 0;
}