Решение задачи IQ тест с Codeforces

#include <bits/stdc++.h>

using namespace std;

int main()
{
    int n, nech = 0, ch = 0, temp;
    cin >> n;
    vector<int> a;
    for(int i = 0; i < n; ++i){
        cin >> temp;
        a.push_back(temp);
    }
    for(int i = 0; i < a.size(); ++i){
        if(a[i] % 2)
            ++nech;
        else
            ++ch;
        if(nech > 1 || ch > 1)
            break;
    }
    if(nech > 1){
        for(int i = 0; i < a.size(); ++i)
            if(a[i] % 2 == 0){
                cout << i + 1;
                return 0;
            }
        }
    else{
         for(int i = 0; i < a.size(); ++i)
            if(a[i] % 2){
                cout << i + 1;
                return 0;
            }
    }

    return 0;
}