Codeforces 158A Next Round

题目链接

http://codeforces.com/problemset/problem/158/A

#include<iostream>
#include<string>
using namespace std;

int main()
{
    int n,k;
    int num=0;
    int arr[100]={0};
    cin>>n>>k;
    for(int i=0;i<n;i++)
    {
        cin>>arr[i];
    }
    int i;
    for(i=0;i<k&&arr[i]!=0;i++)
    {
        num++;
    }
    int flag=1;
    while(arr[i-1]==arr[i]&&arr[i]!=0)
    {
        num++;
        i++;

    }
    cout<<num;

    return 0;
}