Лучший способ разделить вектор на два меньших массива? - PullRequest
Новая
       4

Лучший способ разделить вектор на два меньших массива?

24 голосов
/ 21 марта 2012

Что я пытаюсь сделать:

Я пытаюсь разделить вектор на два отдельных массива. Текущий вектор int содержит элемент на строку в текстовом файле. Текстовый файл представляет собой список случайных целых чисел.

Как я планирую это сделать:

Моя текущая идея - создать два регулярных массива int, затем выполнить итерацию по всему вектору и скопировать n / 2 элементов в каждый из массивов.

Что бы я хотел знать:

Какой самый элегантный способ выполнить мою задачу? У меня такое чувство, что я могу сделать это, не перебирая вектор несколько раз.

Код: 

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

vector<int> ifstream_lines(ifstream& fs)
{
  vector<int> out;
  int temp;
  while(fs >> temp)
  {
    out.push_back(temp);
  }
  return out;
}

vector<int> MergeSort(vector<int>& lines)
{
  int split = lines.size() / 2;
  int arrayA[split];
  int arrayB[split];
}

int main(void) 
{
  ifstream fs("textfile.txt");
  vector<int> lines;
  lines = ifstream_lines(fs);

  return 0;
}

Спасибо:)

Ответы [ 4 ]

59 голосов
/ 21 марта 2012

Использовать итераторы. 

std::vector<int> lines;
// fill
std::size_t const half_size = lines.size() / 2;
std::vector<int> split_lo(lines.begin(), lines.begin() + half_size);
std::vector<int> split_hi(lines.begin() + half_size, lines.end());

Поскольку диапазоны итераторов представляют полуоткрытые диапазоны [begin, end), вам не нужно добавлять 1 во второй начальный итератор: lines.begin() + half_size не копируется в первый вектор.

Обратите внимание, что такие вещи, как 

int split = lines.size() / 2;
int arrayA[split];
int arrayB[split];

не являются стандартными C ++ (и, как таковые, не переносимы). Это так называемые массивы переменной длины (VLA для краткости), и это вещь C99. Некоторые компиляторы имеют их как расширение при компиляции кода C ++ (GCC, Clang). Всегда компилируйте с -pedantic, чтобы получить предупреждение. Эти VLA действуют непривычно для не POD-типов и, как правило, бесполезны, поскольку вы даже не можете их вернуть.

2 голосов
/ 09 марта 2016

Если вы не можете использовать код ответа Xeo из-за строгих правил компилятора или вам нужен более общий способ, попробуйте std::advance: 

#include <vector>
#include <iterator>

size_t middle = input.size()/2;
std::vector<int>::const_iterator middleIter(input.cbegin());
std::advance(middleIter, middle);

std::vector<int> leftHalf(input.begin(), middleIter);
std::vector<int> rightHalf(middleIter, input.end());
2 голосов
/ 22 марта 2012

Если вам нужна только ссылка на числа без манипулирования ими, вы можете сделать: 

int *array_1 = &lines[0];
int *array_2 = &lines[lines.size() / 2];

array_1 и array_2 являются указателями на начало и середину вектора. Это работает, поскольку STL гарантирует, что векторы хранят свои элементы в непрерывной памяти. Обратите внимание, что ссылка на lines.begin () не может быть использована для этого.

1 голос
/ 08 мая 2018

Решение для разделения вектора на части с переменным числом, используя итератор. 

#include <iostream>
#include <vector>

int main()
{
   // Original vector of data
   std::vector<double> mainVec{1.2, 2.3, 3.4, 4.5, 5.6, 6.7, 7.8, 8.9, 9.0};
   // Result vectors
   std::vector<std::vector<double>> subVecs{};
   // Start iterator
   auto itr = mainVec.begin();
   // Variable to control size of non divided elements
   unsigned fullSize = mainVec.size();
   // To regulate count of parts
   unsigned partsCount = 4U;
   for(unsigned i = 0; i < partsCount; ++i)
   {
       // Variable controls the size of a part
       auto partSize = fullSize / (partsCount - i);
       fullSize -= partSize;
       // 
       subVecs.emplace_back(std::vector<double>{itr, itr+partSize});
       itr += partSize;
   }
   // Print out result
   for (const auto& elemOuter : subVecs)
   {
       std::cout << std::fixed;
       for (const auto& elemInner : elemOuter)
       {
           std::cout << elemInner << " ";
       }
       std::cout << "\n";
   }
}
...