Codeforces 158A

Posted 2021-03-05 chopstick8083

#include <stdio.h>
int main()
{
	int n, k;
	scanf("%d %d
", &n, &k);
	int temp;
	int ans=0, com;
	int i;
	for(i=0; i<n; ++i)
	{
		scanf("%d", &temp);
		if(temp>0)
		{
			if(i<(k-1))
				++ans;
			else if(i==(k-1))
			{
				com=temp;
				++ans;
			}
			else if(temp==com)
				++ans;
		}
	}
	printf("%d
", ans);
	return 0;
}

　　@本题的核心代码是位于第12行至第23行的if条件语句。