【CodeForces】[25A]IQ test

#include<stdio.h>
int a[120];
int n;
void solve() {
int cnt1=0,cnt2=0;
for(int i=1; i<=n; i++) {
if(a[i]&1)
cnt1++;
else
cnt2++;
if(cnt1>1&&i>cnt1)
for(int i=1; i<=n; i++)
if(a[i]%2==0) {
printf("%d\n",i);
return ;
}
if(cnt2>1&&i>cnt2)
for(int i=1; i<=n; i++)
if(a[i]&1) {
printf("%d\n",i);
return ;
}
}
return ;
}
int main() {
while(scanf("%d",&n)!=EOF) {
for(int i=1; i<=n; i++)
scanf("%d",&a[i]);
solve();
}
return 0;
}