Codeforces Round #549 (Div. 2) A—C
Three years have passes and nothing changed. It is still raining in London, and Mr. Black has to close all the doors in his home in order to not be flooded. Once, however, Mr. Black became so nervous that he opened one door, then another, then one more and so on until he opened all the doors in his house.
There are exactly two exits from Mr. Black's house, let's name them left and right exits. There are several doors in each of the exits, so each door in Mr. Black's house is located either in the left or in the right exit. You know where each door is located. Initially all the doors are closed. Mr. Black can exit the house if and only if all doors in at least one of the exits is open. You are given a sequence in which Mr. Black opened the doors, please find the smallest index kk such that Mr. Black can exit the house after opening the first kk doors.
We have to note that Mr. Black opened each door at most once, and in the end all doors became open.
The first line contains integer nn (2≤n≤2000002≤n≤200000) — the number of doors.
The next line contains nn integers: the sequence in which Mr. Black opened the doors. The ii-th of these integers is equal to 00 in case the ii-th opened door is located in the left exit, and it is equal to 11 in case it is in the right exit.
It is guaranteed that there is at least one door located in the left exit and there is at least one door located in the right exit.
Print the smallest integer kk such that after Mr. Black opened the first kk doors, he was able to exit the house.
5 0 0 1 0 0
3
4 1 0 0 1
3
In the first example the first two doors are from the left exit, so when Mr. Black opened both of them only, there were two more closed door in the left exit and one closed door in the right exit. So Mr. Black wasn't able to exit at that moment.
When he opened the third door, all doors from the right exit became open, so Mr. Black was able to exit the house.
In the second example when the first two doors were opened, there was open closed door in each of the exit.
With three doors opened Mr. Black was able to use the left exit.
弱智题目 记录最后一个0和1的位置 输出比较小的那个下标就行
#include<bits/stdc++.h> using namespace std; typedef long long ll; const int maxn=2e5+5; const int INF=1e9+5; typedef pair<int,int> pii; int d[maxn],vis[maxn],cnt[maxn],pre[maxn],sum[maxn],val[maxn]; vector<pii> e[maxn]; int n,k; int main() { cin>>n; int a[maxn]; int x=0,y=0,px,py; for(int i=0;i<n;i++) { cin>>a[i]; if(a[i]==0) {x++;px=i;} if(a[i]==1) {y++;py=i;} } if(px>py) cout<<py+1<<endl; else cout<<px+1<<endl; return 0; }