给定一系列正整数,请设计一个尽可能高效的算法,查找倒数第K个位置上的数字。
输入格式:
输入首先给出一个正整数K,随后是若干正整数,最后以一个负整数表示结尾(该负数不算在序列内,不要处理)。
输出格式:
输出倒数第K个位置上的数据。如果这个位置不存在,输出错误信息NULL
。
输入样例:
1
| 4 1 2 3 4 5 6 7 8 9 0 -1
|
输出样例:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
| #include<stdio.h> #include<iostream> #include<string.h> using namespace std; int main() { int k , cnt=0,tmp,flag=0,found = 0; cin>>k; int* num=(int*)malloc(sizeof(int)*k); //只存k个正整数 while(1) { scanf("%d",&tmp); if(tmp<0) //tmp为-1退出循环 break; num[cnt++]=tmp; if(!found&&cnt>=k) found=1; cnt%=k;
} if(found==0) printf("NULL"); else printf("%d",num[cnt]);//输出位于最后输入的元素的下一个元素,因为最后输入的元素就是倒数第一个元素,cnt++后指向它的下一个元素,只存了k个,所以它的下一个就是倒数第k个了 return 0; }
|
在这里只要cnt大于k则一定会找到,found就会变成1不会变回去
关键在于只分配k个大小xxx-10kk+1xxxx