求助

#include<iostream>
#include<algorithm>
using namespace std;
int main(){
	int a_len;
	cin>>a_len;
	int a[a_len],swap[a_len],sum=0;
	for(int i = 0;i <= a_len-1;i ++){
		cin>>a[i];
	}
	for(int i = 0;i <= a_len-1;i ++){
		swap[i] = a[i] % 2;//当所有数取余2时，偶数为0，奇数为1 
	}
	for(int i = 0;i <= a_len-1;i ++){
		if(swap[i] == 0){
			sum ++;
		}
	}
	if(sum = 1){
		for(int i = 0;i <= a_len-1;i ++){
			if(swap[i] == 1){
				cout<<i+1;
			}
		}	
	}
	else{
		for(int i = 0;i <= a_len-1;i ++){
			if(swap[i] == 0){
				cout<<i+1;
			}
		}
	}
}