Accepted solution

158A - Next Round.

Open original problem
Source code

#include <iostream>

#include<vector>

#include<string>

#include<math.h>

#include<set>

#include<map>

#include<algorithm>

/***** S. M. Sabiul Hajjaj Sabit *****/

using namespace std;

template <typename T>long long int power(T a, T b)

{

    long long int pow=1;

    for(int i=0; i<b; i++)

    {

        pow *= a;

    }

    return pow;

}

///*********Write Code From Here*********///

int main()

{

    int n,k;

    int count=0;

    vector<int> arr;

    cin>>n>>k;

    for(int i=0; i<n; i++)

    {

        int t;

        cin>>t;

        arr.push_back(t);

    }

    for(int i=0; i<n; i++)

    {

        if(arr[i]>0 && arr[i]>=arr[k-1])

            count++;

    }

    cout<<count<<endl;

}
Join the discussion

Add your comment