Как заполнить вектор нетривиальными начальными значениями?

Я знаю, как заполнить std::vector нетривиальными начальными значениями, например. порядковые номера:

void IndexArray( unsigned int length, std::vector<unsigned int>& v )
{
    v.resize(length);
    for ( unsigned int i = 0; i < length; ++i )
    {
        v[i] = i;
    }
}

Но это цикл for. Есть ли элегантный способ сделать это с меньшим количеством строк кода с использованием функций stl (и не с помощью Boost)?

Ответы

Ответ 1

Вы можете использовать алгоритм генерации для более общего способа заполнения контейнеров:

#include <iostream>
#include <algorithm>
#include <vector>

struct c_unique {
   int current;
   c_unique() {current=0;}
   int operator()() {return ++current;}
} UniqueNumber;


int main () {
  vector<int> myvector (8);
  generate (myvector.begin(), myvector.end(), UniqueNumber);

  cout << "\nmyvector contains:";
  for (vector<int>::iterator it=myvector.begin(); it!=myvector.end(); ++it)
    cout << " " << *it;

  cout << endl;

  return 0;
}

Это было бесстыдно снято и отредактировано из cplusplusreference.

Ответ 2

Обычно я использую std::generate плюс простой генератор:

template <typename T>
struct gen {
    T x;
    gen(T seed) : x(seed) { }

    T operator ()() { return x++; }
};

generate(a.begin(), a.end(), gen<int>(0));

Ответ 3

Если вы используете SGI STL (или производную, такую как STLPort), вы можете использовать iota.: -)

void IndexArray(unsigned int length, vector<unsigned int>& v)
{
    vector<unsigned int>(length).swap(v);
    iota(v.begin(), v.end(), 0);
}

Ответ 4

Существует также функция iota() в adobe.ASL, (и value_iterator). В boost есть counting_iterator, и я подозреваю, что несколько других способов генерировать числовые последовательности на лету в boost.

Ответ 5

Я знаю, что это уже ответили, но я предпочитаю функцию "заполнить" в библиотеке алгоритмов, так как мне кажется более понятным читать:

// fill algorithm example
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

int main () {
  vector<int> myvector (8);                       // myvector: 0 0 0 0 0 0 0 0

  fill (myvector.begin(),myvector.begin()+4,5);   // myvector: 5 5 5 5 0 0 0 0
  fill (myvector.begin()+3,myvector.end()-2,8);   // myvector: 5 5 5 8 8 8 0 0

  cout << "myvector contains:";
  for (vector<int>::iterator it=myvector.begin(); it!=myvector.end(); ++it)
    cout << " " << *it;

  cout << endl;

  return 0;
}

Это тоже было бесстыдно снято с cplusplusreference.

Ответ 6

Если у вас есть массив стиля C, вы можете использовать std: copy, например,

int c_array[] = {3,4,5};

const int* pbegin = &c_array[0];
const size_t c_array_size = sizeof(c_array) / sizeof(c_array[0]);
const int* pend  = pbegin + c_array_size;

std::vector<int> v;
v.reserve(c_array_size);
std::copy(pbegin, pend, std:back_inserter(v));