I can't understand what is wrong with this piece of code of mine as a solution of the problem 158A:

include

using namespace std; int main() { int n,k,a,cnt=0,b; scanf("%d %d",&n,&k); for(int i=0;i<n;i++) { scanf("%d",&a); if(i==k) b==a; if(a>0 && cnt<k) cnt++; else { if(b==a & a>0) cnt++; } } printf("%d\n",cnt); return 0; } Then I submitted this solution and it worked:

include

using namespace std; int main() { int n,k,a,b; scanf("%d %d",&n,&k); for (int i = 0; i < k; ++i) { scanf("%d",&a); if (!a) { printf("%d\n",i); return 0; } } for (int i = k; i < n; ++i) { scanf("%d",&b); if (b != a) { printf("%d\n",i); return 0; } } printf("%d\n",n);
return 0; } Please help me understand why the code that I submitted earlier fails.