1143A - The Doors

#include <iostream>

#include<vector>

#include<string>

#include<math.h>

#include<set>

#include<map>

#include<stack>

#include<queue>

#include<algorithm>

/* ==================================================

  |                                                  |

  |    Author: *** S. M. Sabiul Hajjaj Sabit ***     |

  |       Last Edit:  May,14,2019                    |

   ================================================== */

///***global declaration***///

#define all(arr) arr.begin(),arr.end()

#define pb push_back

#define sz(x) x.size()

#define min3(a,b,c)  min(min(a,b),c)

#define max3(a,b,c)  max(max(a,b),c)

using namespace std;

int i,n,m,sum=0,ans=0;

template <typename T>long long int power(T a, T b)

{

    long long int pow=1;

    for(int i=0; i<b; i++)

    {

        pow *= a;

    }

    return pow;

}

///*********Write Code From Here*********///

int main()

{

    int n;

    cin>>n;

    vector<int> arr;

    int left=0,right=0;

    for(int i=0; i<n; i++)

    {

        int t;

        cin>>t;

        if(t==0)left++;

        else right++;

        arr.pb(t);

    }

    for(int i=0; i<sz(arr); i++)

    {

        if(arr[i]==0)left--;

        else right--;

        if(left==0 || right==0)

        {

            cout<<i+1<<endl;

            return 0;

        }

    }

}