Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- /*******************************************************************************\
- |*********S*** ****** ************* ************* ************** * ******|
- |*********** ******** ************ ************ *********** ** ******|
- |********* ********** *********** *** *********** * ********* *** ******|
- |******** *********** *I******** **** ********** *** ***** ***** ******|
- |******** *********** ********* ****** ********* ***** ** ****** ******|
- |********** ********* ******** ***A**** ******** ******* ******** ******|
- |*********** ******** ******* ********** ******* ***************** ******|
- |********** ********* ****** ************ ****** ***************** ******|
- |******** *********** ***** ************** ***** ********M******** ******|
- |***** ************** **** **************** **** ***************** ******|
- |*******************************************************************************|
- |**************BSMRSTU********************************CSE***********************|
- \*******************************************************************************/
- #include<bits/stdc++.h>
- #define ll long long
- #define B break
- #define C continue
- #define sf scanf
- #define pf printf
- #define byebye return 0
- using namespace std;
- int main()
- {
- int n,a,i,j,k,t=0;
- cin>>n>>a;
- a--;
- int c[n];
- for(i=0; i<n; i++)
- {
- cin>>c[i];
- }
- if(c[a]==1)t++;
- if((a)<=(n/2))
- {
- for(i=a-1,j=a+1; i>=0, j<n; i--,j++)
- {
- if(c[i]==1 && c[j]==1)
- t+=2;
- if(i==-1 && j>2*a)
- {
- for(k=j; k<n; k++)
- {
- if(c[k]==1)t++;
- }
- break;
- }
- }
- }
- else if(a>(n/2))
- {
- for(i=a-1,j=a+1; i>=0, j<n; i--,j++)
- {
- if(c[i]==1 && c[j]==1)
- t+=2;
- }
- for(i=((2*a)-n); i>=0; i--)
- {
- if(c[i]==1)t++;
- }
- }
- cout<<t;
- byebye;
- }
Advertisement
Add Comment
Please, Sign In to add comment