MT 名次并列

有 n个人参加比赛，其中分数排名前  的人将被选入下一轮(选入下一轮的人分数必须为正)，特别的，如果几个人分数相同且刚好并列处于第 k名(或是并列k -i名，但是全部算入后选入下一轮的人数超过  人)，这几个人都将被选入下一轮。
小码哥请你输出进入下一轮的人数?

格式
输入格式:输入一共两行，第一行是两个数n,k (1<=k<=n<=1e5
第二行有n个数，分别参加比赛的人的分数 a,a2,...,an(0<=ai<=100)。
输出格式:输出进入下一轮的人数。 

样例 1
输入:8 5

        7 5 10 7 9 8 7 5
输出:6

#include<stdio.h>
#include<stdlib.h>
#define N 100007
int n,k,a[N],ans; // 定义变量n、k、a、ans，其中a为数组，N为数组大小的最大值
int cmp(const void *a, const void *b) { // 快速排序函数定义
    int x=*(const int*)a,y=*(const int*)b; // 强制类型转换，将指针a和指针b转换成整数类型，并赋值给x和y
    if(x>y)
        return -1; // 如果x>y，则返回-1
    else if(x<y)
        return 1; // 如果x<y，则返回1
    else
        return 0; // 否则返回0
}
int main()
{
    scanf("%d%d",&n,&k); // 输入n和k的值
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]); // 输入n个数并存储到数组a中
    qsort(a+1,n,sizeof(int),cmp); // 使用快速排序函数对数组a进行降序排列
    for(int i=1;i<=k;i++){ // 遍历前k个数
        if(a[i]>0)
            ans++; // 如果第i个数大于0，则计数器加1
        else
        {
            printf("%d",ans); // 否则输出计数器的值，并结束程序
            return 0;
        }
    }
    for(int i=k+1;i<=n;i++) // 遍历剩余的数
        if(a[i]==a[k])
            ans++; // 如果第i个数和第k个数相等，则计数器加1
    printf("%d",ans); // 输出计数器的值，并结束程序
    return 0;
}