fork(1) download
copy 
  1. #include <iostream>
  2. #include <cstdio>
  3. using namespace std;
  4.  
  5. int main() {
  6. 	int n, k, a[55];
  7. 	scanf("%d%d", &n, &k);
  8. 	for(int i=0; i<n; i++) scanf("%d", &a[i]);
  9. 	if(a[k-1]>0) {
  10. 		for(int i=k; i<n; i++) {
  11. 			if(a[i]==a[k-1]) k++;
  12. 			else break;
  13. 		}
  14. 	}
  15. 	else {
  16. 		for(int i=k-1; i>=0; i--) {
  17. 			if(a[i]==0) k--;
  18. 			else break;
  19. 		}
  20. 	}
  21. 	printf("%d\n", k);
  22. 	return 0;
  23. }
Success #stdin #stdout 0s 3344KB
comments (?)
stdin
copy 
4 2
0 0 0 0
stdout
copy 
0
https://ideone.com/e9JrhL
language:
C++ (gcc 8.3)
created:
11 years ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!