Tipus de comarques - X41648

#include <iostream>
#include <vector>
using namespace std;

struct Poblacio {
  string pob;
  int    alt;
  string com;
};

string tipusComarca(const vector<Poblacio> & lpob, const string & comarca) {
    // Add code here ...
    int o=0;
    vector<int> estat;
    for(int i=0;i<lpob.size();++i){
        if(lpob[i].com==comarca){
            estat.push_back(lpob[i].alt);
            ++o;
        }
    }
    if(estat[0]<500){
        for(int i=1;i<o;++i){
            if(estat[i]>=500) return "mixed";
        }
        return "seaside";
    }
    else if(estat[0]>=500){
        for(int i=1;i<o;++i){
            if(estat[i]<500) return "mixed";
        }
        return "mountain";
    }
}

vector<Poblacio> llegir_poblacions(int n) {
  vector<Poblacio> lpobl(n);
  for (int i = 0; i < n; ++i) {
    cin >> lpobl[i].pob >> lpobl[i].alt >> lpobl[i].com;
  }
  return lpobl;
}

int main() {
    // Add code here ...
    int n;
    cin>>n;
    vector<Poblacio> e=llegir_poblacions(n);
    string comarca;
    while(cin>>comarca){
        cout<<comarca<<": "<<tipusComarca(e,comarca)<<endl;
     }
}