158A/main.cpp


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
26
27
28

#include <iostream>
#include <vector>

using namespace std;
int main(){
        int a,b;
        vector<int> v;
        cin>>a>>b;
        int c;

        for(int i = 0; i < a; i++){
                cin>>c;
                v.push_back(c);
        } 
       
        int k = v.at(b - 1);
        int count = 0;

        for(int i = 0; i < v.size(); i++){
                if(v.at(i) >= k && v.at(i) > 0){
                        count++;
                }
        }
        cout<<count;


        return 0;
}