STR - DDD

#include <iostream>
#include <cstdio>
#include <bits/stdc++.h>

using namespace std;

int main()
{
    int i;                  // contador
    int ddd = 0;            // ddd da cidade
    string cidade;          // nome da cidade
    int listaDDD[9] = {61, 71, 11, 21, 32, 19, 27, 31, 35};
    string listaCidades[9] = {"Brasilia", "Salvador", "Sao Paulo", "Rio de Janeiro",
                              "Juiz de Fora", "Campinas", "Vitoria", "Belo Horizonte",
                              "Santa Rita do Sapucai"};
    // lendo cidade
    getline(cin, cidade);
    // buscando ddd da cidade
    for(i = 0; i < 9; i++)
    {
        if(cidade == listaCidades[i]) {
            ddd = listaDDD[i];
            printf("%d\n", ddd);
            return 0;
        }
    }
    // retornando aviso quando nao encontrar a cidade
    printf("DDD nao cadastrado\n");
    return 0;
}