Использование рекурсии и backtracking для генерации всех возможных комбинаций
Я пытаюсь реализовать класс, который будет генерировать все возможные неупорядоченные N-кортежи или комбинации с учетом количества элементов и размера комбинации.
другими словами, при вызове этого:
NTupleUnordered unordered_tuple_generator(3, 5, print);
unordered_tuple_generator.Start();
print () - функция обратного вызова, установленная в конструкторе. Вывод должен быть:
{0,1,2}
{0,1,3}
{0,1,4}
{0,2,3}
{0,2,4}
{0,3,4}
{1,2,3}
{1,2,4}
{1,3,4}
{2,3,4}
это то, что у меня есть до сих пор:
class NTupleUnordered {
public:
NTupleUnordered( int k, int n, void (*cb)(std::vector<int> const&) );
void Start();
private:
int tuple_size; //how many
int set_size; //out of how many
void (*callback)(std::vector<int> const&); //who to call when next tuple is ready
std::vector<int> tuple; //tuple is constructed here
void add_element(int pos); //recursively calls self
};
и это реализация рекурсивной функции, Start () - это просто удар функция запуска, чтобы иметь более чистый интерфейс, вызывает только add_element (0);
void NTupleUnordered::add_element( int pos )
{
// base case
if(pos == tuple_size)
{
callback(tuple); // prints the current combination
tuple.pop_back(); // not really sure about this line
return;
}
for (int i = pos; i < set_size; ++i)
{
// if the item was not found in the current combination
if( std::find(tuple.begin(), tuple.end(), i) == tuple.end())
{
// add element to the current combination
tuple.push_back(i);
add_element(pos+1); // next call will loop from pos+1 to set_size and so on
}
}
}
если бы я хотел создать все возможные комбинации постоянного размера N, скажем, комбинации размера 3, которые я мог бы сделать:
for (int i1 = 0; i1 < 5; ++i1)
{
for (int i2 = i1+1; i2 < 5; ++i2)
{
for (int i3 = i2+1; i3 < 5; ++i3)
{
std::cout << "{" << i1 << "," << i2 << "," << i3 << "}n";
}
}
}
если N не является константой, вам нужна рекурсивная функция, имитирующая вышеуказанное функция, выполняя каждый for-loop в своем собственном фрейме. Когда цикл завершается, программа возвращается к предыдущему кадру, другими словами, отступает.
Я всегда были проблемы с рекурсией, и теперь мне нужно объединить ее с обратным отслеживанием, чтобы сгенерировать все возможные комбинации. Какие-нибудь указания на то, что я делаю неправильно? Что я должен делать, или я упускаю из виду?
P. S: это задание колледжа, которое также включает в себя в основном то же самое для упорядоченных N-кортежей.
спасибо заранее!
/////////////////////////////////////////////////////////////////////////////////////////
просто хотел следуйте с правильным кодом на всякий случай, если кто-то еще там задается тем же вопросом.
void NTupleUnordered::add_element( int pos)
{
if(static_cast<int>(tuple.size()) == tuple_size)
{
callback(tuple);
return;
}
for (int i = pos; i < set_size; ++i)
{
// add element to the current combination
tuple.push_back(i);
add_element(i+1);
tuple.pop_back();
}
}
и для случая упорядоченных N-кортежей:
void NTupleOrdered::add_element( int pos )
{
if(static_cast<int>(tuple.size()) == tuple_size)
{
callback(tuple);
return;
}
for (int i = pos; i < set_size; ++i)
{
// if the item was not found in the current combination
if( std::find(tuple.begin(), tuple.end(), i) == tuple.end())
{
// add element to the current combination
tuple.push_back(i);
add_element(pos);
tuple.pop_back();
}
}
}
спасибо Джейсон за ваш тщательный ответ!
3 ответов
хороший способ подумать о формировании N комбинаций-посмотреть на структуру как на дерево комбинаций. Пересечение этого дерева становится естественным способом подумать о рекурсивной природе алгоритма, который вы хотите реализовать, и о том, как будет работать рекурсивный процесс.
предположим, например, что у нас есть последовательность, {1, 2, 3, 4}
, и мы хотим найти все 3-комбинации в этом наборе. "Дерево" комбинаций будет выглядеть так: следующий:
root
________|___
| |
__1_____ 2
| | |
__2__ 3 3
| | | |
3 4 4 4
переход от корня с использованием обхода предварительного порядка и определение комбинации, когда мы достигаем листового узла, мы получаем комбинации:
{1, 2, 3}
{1, 2, 4}
{1, 3, 4}
{2, 3, 4}
таким образом, в основном идея заключалась бы в последовательности через массив, используя значение индекса, что для каждого этапа нашей рекурсии (который в этом случае был бы "уровнями" дерева), инкременты в массив, чтобы получить значение, которое будет включено в набор комбинаций. Также обратите внимание, что мы только надо повторить N раз. Поэтому у вас будет некоторая рекурсивная функция, подпись которой будет выглядеть примерно так:
void recursive_comb(int step_val, int array_index, std::vector<int> tuple);
здесь step_val
указывает, как далеко мы должны рекурсировать,array_index
значение сообщает нам, где мы находимся в наборе, чтобы начать добавлять значения в tuple
и tuple
, как только мы закончим, будет экземпляр комбинации в наборе.
вам нужно будет позвонить recursive_comb
из другой нерекурсивной функции это в основном "запускает" рекурсивный процесс, инициализируя tuple
вектор и ввод максимальных рекурсивных шагов (т. е. количество значений, которые мы хотим в кортеж):
void init_combinations()
{
std::vector<int> tuple;
tuple.reserve(tuple_size); //avoids needless allocations
recursive_comb(tuple_size, 0, tuple);
}
наконец-то recusive_comb
функция будет выглядеть примерно так:
void recursive_comb(int step_val, int array_index, std::vector<int> tuple)
{
if (step_val == 0)
{
all_combinations.push_back(tuple); //<==We have the final combination
return;
}
for (int i = array_index; i < set.size(); i++)
{
tuple.push_back(set[i]);
recursive_comb(step_val - 1, i + 1, tuple); //<== Recursive step
tuple.pop_back(); //<== The "backtrack" step
}
return;
}
вы можете увидеть рабочий пример этого кода здесь:http://ideone.com/78jkV
обратите внимание, что это не быстрый версия алгоритма, в что мы берем некоторые дополнительные ветви, которые нам не нужны, которые создают ненужные вызовы копирования и функций и т. д. ... но, надеюсь, он пересекается с общей идеей рекурсии и возврата, и как они работают вместе.
лично я бы пошел с простым итеративным решением.
представляют набор узлов как набор битов. Если вам нужно 5 узлов, то есть 5 бит, каждый бит представляет определенный узел. Если вы хотите 3 из них в своем tupple, вам просто нужно установить 3 бита и отслеживать их местоположение.
в основном это простая вариация на fonding все разные подмножества комбинаций узлов. Где классическая реализация представляет набор узлов как целое число. Каждый бит в integer представляет собой узел. Пустой набор равен 0. Затем вы просто увеличиваете целое число, каждое новое значение - это новый набор узлов (битовый шаблон, представляющий набор узлов). Просто в этом варианте вы убедитесь, что всегда есть 3 узла.
просто чтобы помочь мне думать, что я начинаю с 3 верхних узлов active { 4, 3, 2 }. Потом я начинаю обратный отсчет. Но было бы тривиально изменить это, чтобы рассчитывать в другом направлении.
#include <boost/dynamic_bitset.hpp>
#include <iostream>
class TuppleSet
{
friend std::ostream& operator<<(std::ostream& stream, TuppleSet const& data);
boost::dynamic_bitset<> data; // represents all the different nodes
std::vector<int> bitpos; // tracks the 'n' active nodes in the tupple
public:
TuppleSet(int nodes, int activeNodes)
: data(nodes)
, bitpos(activeNodes)
{
// Set up the active nodes as the top 'activeNodes' node positions.
for(int loop = 0;loop < activeNodes;++loop)
{
bitpos[loop] = nodes-1-loop;
data[bitpos[loop]] = 1;
}
}
bool next()
{
// Move to the next combination
int bottom = shiftBits(bitpos.size()-1, 0);
// If it worked return true (otherwise false)
return bottom >= 0;
}
private:
// index is the bit we are moving. (index into bitpos)
// clearance is the number of bits below it we need to compensate for.
//
// [ 0, 1, 1, 1, 0 ] => { 3, 2, 1 }
// ^
// The bottom bit is move down 1 (index => 2, clearance => 0)
// [ 0, 1, 1, 0, 1] => { 3, 2, 0 }
// ^
// The bottom bit is moved down 1 (index => 2, clearance => 0)
// This falls of the end
// ^
// So we move the next bit down one (index => 1, clearance => 1)
// [ 0, 1, 0, 1, 1]
// ^
// The bottom bit is moved down 1 (index => 2, clearance => 0)
// This falls of the end
// ^
// So we move the next bit down one (index =>1, clearance => 1)
// This does not have enough clearance to move down (as the bottom bit would fall off)
// ^ So we move the next bit down one (index => 0, clearance => 2)
// [ 0, 0, 1, 1, 1]
int shiftBits(int index, int clerance)
{
if (index == -1)
{ return -1;
}
if (bitpos[index] > clerance)
{
--bitpos[index];
}
else
{
int nextBit = shiftBits(index-1, clerance+1);
bitpos[index] = nextBit-1;
}
return bitpos[index];
}
};
std::ostream& operator<<(std::ostream& stream, TuppleSet const& data)
{
stream << "{ ";
std::vector<int>::const_iterator loop = data.bitpos.begin();
if (loop != data.bitpos.end())
{
stream << *loop;
++loop;
for(; loop != data.bitpos.end(); ++loop)
{
stream << ", " << *loop;
}
}
stream << " }";
return stream;
}
главное тривиально:
int main()
{
TuppleSet s(5,3);
do
{
std::cout << s << "\n";
}
while(s.next());
}
вывод:
{ 4, 3, 2 }
{ 4, 3, 1 }
{ 4, 3, 0 }
{ 4, 2, 1 }
{ 4, 2, 0 }
{ 4, 1, 0 }
{ 3, 2, 1 }
{ 3, 2, 0 }
{ 3, 1, 0 }
{ 2, 1, 0 }
версия shiftBits () с использованием цикла
int shiftBits()
{
int bottom = -1;
for(int loop = 0;loop < bitpos.size();++loop)
{
int index = bitpos.size() - 1 - loop;
if (bitpos[index] > loop)
{
bottom = --bitpos[index];
for(int shuffle = loop-1; shuffle >= 0; --shuffle)
{
int index = bitpos.size() - 1 - shuffle;
bottom = bitpos[index] = bitpos[index-1] - 1;
}
break;
}
}
return bottom;
}
в MATLAB:
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% combinations.m
function combinations(n, k, func)
assert(n >= k);
n_set = [1:n];
k_set = zeros(k, 1);
recursive_comb(k, 1, n_set, k_set, func)
return
function recursive_comb(k_set_index, n_set_index, n_set, k_set, func)
if k_set_index == 0,
func(k_set);
return;
end;
for i = n_set_index:length(n_set)-k_set_index+1,
k_set(k_set_index) = n_set(i);
recursive_comb(k_set_index - 1, i + 1, n_set, k_set, func);
end;
return;
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
Test:
>> combinations(5, 3, @(x) printf('%s\n', sprintf('%d ', x)));
3 2 1
4 2 1
5 2 1
4 3 1
5 3 1
5 4 1
4 3 2
5 3 2
5 4 2
5 4 3