Минимизировать количество дорог, которые придётся проехать от заселённого микрорайона до участка

#include <iostream>
#include <vector>
#include <stack>
 
// Узел дерева/микрорайон
struct tree_node
{
    bool type; // тип микрорайона
    std::vector<tree_node*> near; // микрорайоны соединённые дорогой с текущим
};
 
// Вспомогательная структура для поиска высоты дерева
class trace
{
public:
    const tree_node* node; // узел
    const tree_node* parent; // родитель
    int processed; // обработанных подузлов
    int height; // высота
 
    // Конструктор для упрощения stack.emplace
    trace(const tree_node* node, const tree_node* parent)
    {
        this->node = node;
        this->parent = parent;
        this->processed = 0;
        this->height = node->type - 1;
    }
 
    // Следующий подузел
    const tree_node* next()
    {
        // Пропуск родителя
        if (processed < node->near.size() && node->near[processed] == parent)
            processed++;
        // Подузел или nullptr
        if (processed < node->near.size())
            return node->near[processed++];
        else
            return nullptr;
    }
};
 
// Высота поддерева. -1 если в нём нет населённых микрорайонов
int height(const tree_node* root, const tree_node* parent)
{
    std::stack<trace> backtrace;
    backtrace.emplace(root, parent);
    int max = -1;
    // Пока не проверено всё поддерево
    while (backtrace.size())
    {
        trace& cur = backtrace.top();
        // Обновление максимальной высоты
        if (max > cur.height)
        {
            cur.height = max;
            max = -1;
        }
        // Добавление подузла или извлечение высоты узла
        const tree_node* next = cur.next();
        if (next)
        {
            backtrace.emplace(next, cur.node);
        }
        else
        {
            max = cur.height + (cur.height != -1);
            backtrace.pop();
        }
    }
    return max;
}
 
// Диаметр дерева без ненаселённых микрорайонов
int diameter(const tree_node* root)
{
    int max1 = 0, max2 = 0;
    for (int i = 0; i < root->near.size(); i++)
    {
        int tmp = height(root->near[i], root);
        if (tmp > max1)
        {
            max2 = max1;
            max1 = tmp;
        }
        else if (tmp > max2)
        {
            max2 = tmp;
        }
    }
    return max1 + max2;
}
 
void solution(std::istream& in, std::ostream& out)
{
    int tmp;
    // Кол-во узлов
    in >> tmp;
    std::vector<tree_node> nodes(tmp);
    // Считывание типа узлов и поиск потенциального корня
    for (int i = 0; i < nodes.size(); i++)
    {
        in >> nodes[i].type;
        if (nodes[i].type)
            tmp = i;
    }
    // Считывание связей
    for (int i = 1; i < nodes.size(); i++)
    {
        int a, b;
        in >> a >> b;
        a--, b--;
        nodes[a].near.push_back(&nodes[b]);
        nodes[b].near.push_back(&nodes[a]);
    }
    // Ответ
    tmp = diameter(&nodes[tmp]);
    out << tmp / 2 + tmp % 2;
}
 
int main()
{
    // Вынесено в отдельную функцию чтобы можно было заменить на работу с файлами
    solution(std::cin, std::cout);
}