Нахождение нескольких элементов max в векторе C ++ - PullRequest
Новая
       12

Нахождение нескольких элементов max в векторе C ++

2 голосов
/ 21 апреля 2020

Итак .. Я пытаюсь найти максимальное значение вектора и его положение в векторе. Я использую для L oop, и он работает нормально. Моя проблема в том, что если максимальное значение появляется более одного раза, я хочу знать все позиции, в которых оно появляется в векторе .. Итак, как мне это сделать? Пока что это код, который я использую: (вектор с именем v имеет элементы, которые я читаю из файла, но я не буду добавлять эту часть кода) 

std::vector<double>v;
double maxvalue;
int position=0;
maxvalue = v[0];

for (unsigned int i=0; i<v.size(); i++){
    if (v[i]> maxvalue){
        maxvalue=v[i];
        position= i;
    }
}

Ответы [ 4 ]

3 голосов
/ 21 апреля 2020

Вы можете изменить свой подход, чтобы сохранить вектор индексов с максимальным значением 

#include <cfloat>
#include <iostream>
#include <utility>
#include <vector>

std::pair<double, std::vector<std::size_t>> FindMaxElements(std::vector<double> const& v)
{
    std::vector<std::size_t> indices;
    double current_max = -DBL_MAX;

    for (std::size_t i = 0; i < v.size(); ++i)
    {
        if (v[i] > current_max)
        {
            current_max = v[i];
            indices.clear();
        }

        if (v[i] == current_max)
        {
            indices.push_back(i);
        }
    }

    return std::make_pair(current_max, indices);
}

int main()
{
    auto result = FindMaxElements({1, 4, 7, 2, 7, 3});
    std::cout << "max: " << result.first << '\n';
    std::cout << "indices: ";
    for (auto i : result.second)
         std::cout << i << ' ';
}

Выход 

max: 7
indices: 2 4
2 голосов
/ 21 апреля 2020

Вот двухпроходная версия, использующая стандартную библиотеку (тогда как она может быть чище без нее): 

#include <vector>
#include <algorithm>

int main()
{
    std::vector<double> v {/* fill it*/ };
    std::vector<int> pos;

    auto it = std::max_element(std::begin(v), std::end(v));
    while (it != std::end(v))
    {
        pos.push_back(std::distance(std::begin(v), it));
        it = std::find(std::next(it), std::end(v), *it);
    }
    //...
}
1 голос
/ 21 апреля 2020

Шаблон функции ниже, find_maximums(), возвращает std::vector<size_t>, который содержит позиции, где максимумы находятся во входном векторе. Обратите внимание, что он возвращает пустой вектор индексов, если входной вектор пуст. 

template<typename T>
auto find_maximums(const std::vector<T>& v) {
   std::vector<size_t> indexes;

   for (auto it_max = std::max_element(v.begin(), v.end()); it_max != v.end();
        it_max = std::find(it_max+1, v.end(), *it_max))
   {
      auto index = std::distance(v.begin(), it_max);
      indexes.push_back(index);
   }

   return indexes;
}

В качестве примера использования: 

auto main() -> int {
   std::vector<int> v = {11, 7, 3, 11, 0, 7, 1, 11, 11};
   auto max_indexes = find_maximums(v);

   if (max_indexes.empty())
      return 1;

   std::cout << "max: " << v[max_indexes.front()] << std::endl;
   std::cout << "max at positions: ";
   for (auto idx: max_indexes)
      std::cout << idx << ' ';
   std::cout << '\n';
}

Он выводит: 

max: 11
max at positions: 0 3 7 8
0 голосов
/ 21 апреля 2020

Мои два цента, минуя пару итераторов и компаратор 

template <class It,
          class Comp = std::less<typename std::iterator_traits<It>::value_type>>
auto max_elements_indices(It first, It last, Comp cmp = Comp{})
{
    // This function returns a vector of indices, so to get the maximum, the caller
    // should first check if the returned vector is empty and then use one of
    // those indices to retrieve the value.
    std::vector<std::size_t> indices;
    if (first == last)
        return indices;

    // Using the first element instead of a sentinel value is easier to generalize
    indices.push_back(0);
    auto value = *first;

    for (auto i = std::next(first); i != last; ++i)
    {
        // The most common case should be an element below the maximum
        if ( cmp(*i, value) )
            continue;
        else
        {
            if ( cmp(value, *i) )
            {
                value = *i;
                indices.clear();
            }
            indices.push_back(std::distance(first, i));
        }
    }
    return indices;
}

Тестируемый здесь .

...