796A - Buying A House - CodeForces Solution

#include<bits/stdc++.h>
using namespace std;

#define int long long int

signed main(){
ios::sync_with_stdio(0);
cin.tie(0);
int n,m,k;
cin>>n>>m>>k;
int a[n];
for(int i=0;i<n;i++){
    cin>>a[i];

} 
int ans=-1;
if((m-1)==0 || m==n){
    if(m-1==0){
        int pri=m;
        while(pri!=n){
            if(a[pri]!=0 && a[pri]<=k){ans=pri;
                break;
            }
            else pri++;
        }
          cout<<(abs(m-ans)+1)*10;
    return 0;
    }
    else if(m==n){
        int pri=m-2;
         while(pri>=0){
            if(a[pri]!=0 && a[pri]<=k){
                ans=pri;
                break;
            }
            else pri--;
        }
    }
    cout<<(abs(m-ans)-1)*10;
    return 0;
}

else{
int st=m-2,en=m;
   while(st>=0 || en!=n){
       if(a[st]!=0 && a[st]<=k){
            ans=st;
            break;
       }
       if(a[en]!=0 && a[en]<=k){
        ans=en;
        break;
       }
       else {
        if(st!=0)st--;
        if(en!=n)en++;
       }
   }

cout<<abs(m-ans-1)*10;
}

}