Поиск декартового произведения с ассоциативными массивами PHP
говорят, что у меня есть массив следующим образом:
Array
(
[arm] => Array
(
[0] => A
[1] => B
[2] => C
)
[gender] => Array
(
[0] => Female
[1] => Male
)
[location] => Array
(
[0] => Vancouver
[1] => Calgary
)
)
как найти декартово произведение при сохранении ключей внешнего ассоциативного массива и использовании их во внутренних? Результатом работы алгоритма должно быть следующее:
Array
(
[0] => Array
(
[arm] => A
[gender] => Female
[location] => Vancouver
)
[1] => Array
(
[arm] => A
[gender] => Female
[location] => Calgary
)
[2] => Array
(
[arm] => A
[gender] => Male
[location] => Vancouver
)
...etc.
Я просмотрел довольно много алгоритмов декартовых продуктов, но я застрял на специфике того, как сохранить ассоциативные ключи. Текущий алгоритм я использую дает численные показатели только:
$result = array();
foreach ($map as $a) {
if (empty($result)) {
$result = $a;
continue;
}
$res = array();
foreach ($result as $r) {
foreach ($a as $v) {
$res[] = array_merge((array)$r, (array)$v);
}
}
$result = $res;
}
print_r($result);
любая помощь будет оценили.
10 ответов
вот решение, которое я не постыдился бы показать.
обоснование
предположим, что у нас есть входной массив $input
С N
суб-массивы, как в вашем примере. Каждый
суб-массив имеет Cn
элементов, где n
его индекс внутри $input
, и его ключ Kn
. Я буду ссылаться на i
й пункт n
- й под-массив as Vn,i
.
приведенный ниже алгоритм может быть доказан для работы (за исключением ошибок) индукцией:
1) для N = 1, декартово произведение просто array(0 => array(K1 => V1,1), 1 => array(K1 => V1,2), ... )
-- C1 элементов в общей сложности. Это можно сделать с помощью простого foreach
.
2) Предположим, что $result
уже содержит декартово произведение первых N-1 суб-массивы. Декартово произведение $result
и N-й под-массив может быть получен таким образом:
3) в каждом элементе (массиве) внутри $product
добавить значением KN => VN,1
. Запомните полученный элемент (с добавленной стоимостью); я буду называть его $item
.
4а) Для каждого массива внутри $product
:
4b) для каждого значения в наборе VN,2 ... VN,CN
добавьте к $product
копия $item
, но измените значение с помощью ключа KN
to VN,m
(для всех 2 <= m <= CN
).
две итерации 4a (over $product
) и 4b (над N-М входным суб-массивом) заканчиваются $result
С CN
предметы для каждого элемента, который он имел до итераций, так что в конце $result
действительно содержит декартово произведение первого N sub матрицы.
поэтому алгоритм будет работать для любого Н.
код
function cartesian($input) {
$result = array();
while (list($key, $values) = each($input)) {
// If a sub-array is empty, it doesn't affect the cartesian product
if (empty($values)) {
continue;
}
// Seeding the product array with the values from the first sub-array
if (empty($result)) {
foreach($values as $value) {
$result[] = array($key => $value);
}
}
else {
// Second and subsequent input sub-arrays work like this:
// 1. In each existing array inside $product, add an item with
// key == $key and value == first item in input sub-array
// 2. Then, for each remaining item in current input sub-array,
// add a copy of each existing array inside $product with
// key == $key and value == first item of input sub-array
// Store all items to be added to $product here; adding them
// inside the foreach will result in an infinite loop
$append = array();
foreach($result as &$product) {
// Do step 1 above. array_shift is not the most efficient, but
// it allows us to iterate over the rest of the items with a
// simple foreach, making the code short and easy to read.
$product[$key] = array_shift($values);
// $product is by reference (that's why the key we added above
// will appear in the end result), so make a copy of it here
$copy = $product;
// Do step 2 above.
foreach($values as $item) {
$copy[$key] = $item;
$append[] = $copy;
}
// Undo the side effecst of array_shift
array_unshift($values, $product[$key]);
}
// Out of the foreach, we can add to $results now
$result = array_merge($result, $append);
}
}
return $result;
}
использование
$input = array(
'arm' => array('A', 'B', 'C'),
'gender' => array('Female', 'Male'),
'location' => array('Vancouver', 'Calgary'),
);
print_r(cartesian($input));
посмотреть его в действии!
вот оптимизированная версия декартовой функции @Jon:
function cartesian($input) {
$result = array(array());
foreach ($input as $key => $values) {
$append = array();
foreach($result as $product) {
foreach($values as $item) {
$product[$key] = $item;
$append[] = $product;
}
}
$result = $append;
}
return $result;
}
подробнее о математике этого алгоритма:http://en.wikipedia.org/wiki/Cartesian_product
см. дополнительные примеры этого алгоритма на разных языках:https://rosettacode.org/wiki/Cartesian_product_of_two_or_more_lists
вот что я мог бы придумать:
function inject($elem, $array) {
return array_map(function ($n) use ($elem) { return array_merge((array)$elem, (array)$n); }, $array);
}
function zip($array1, $array2) {
return array_reduce($array1, function ($v, $n) use ($array2) { return array_merge($v, inject($n, $array2)); }, array());
}
function cartesian_product($array) {
$keys = array_keys($array);
$prod = array_shift($array);
$prod = array_reduce($array, 'zip', $prod);
return array_map(function ($n) use ($keys) { return array_combine($keys, $n); }, $prod);
}
(используя псевдо-массив / список / обозначение словаря ниже, так как PHP просто слишком многословен для таких вещей.)
на inject
функция преобразует a, [b]
на [(a,b)]
, т. е. он вводит одно значение в каждое значение массива, возвращая массив массивов. Неважно a
или b
уже является массивом, он всегда будет возвращать двумерный матрица.
inject('a', ['foo', 'bar'])
=> [('a', 'foo'), ('b', 'bar')]
на применение inject
функции для каждого элемента в массиве.
zip(['a', 'b'], ['foo', 'bar'])
=> [('a', 'foo'), ('a', 'bar'), ('b', 'foo'), ('b', 'bar')]
обратите внимание, что это фактически производит декартово произведение, поэтому zip
немного неправильное название. Простое применение этой функции ко всем элементам в наборе данных последовательно дает декартовое произведение для массива любой длины.
zip(zip(['a', 'b'], ['foo', 'bar']), ['42', '76'])
=> [('a', 'foo', '42'), ('a', 'foo', '76'), ('a', 'bar', '42'), …]
это не содержит ключей, но так как элементы все в порядке в результирующем наборе, вы можно просто повторно ввести ключи в результат.
array_combine(['key1', 'key2', 'key3'], ['a', 'foo', '42'])
=> [ key1 : 'a', key2 : 'foo', key3 : '42' ]
применение этого ко всем элементам в продукте дает желаемый результат.
вы можете свернуть вышеуказанные три функции в один длинный оператор, если хотите (что также очистит неправильные имена).
"развернутая" версия без анонимных функций для PHP
function inject($elem, $array) {
$elem = (array)$elem;
foreach ($array as &$a) {
$a = array_merge($elem, (array)$a);
}
return $array;
}
function zip($array1, $array2) {
$prod = array();
foreach ($array1 as $a) {
$prod = array_merge($prod, inject($a, $array2));
}
return $prod;
}
function cartesian_product($array) {
$keys = array_keys($array);
$prod = array_shift($array);
$prod = array_reduce($array, 'zip', $prod);
foreach ($prod as &$a) {
$a = array_combine($keys, $a);
}
return $prod;
}
почему бы не использовать рекурсивный генератор ... проблемы с памятью: близко к none
(и это прекрасно)
function cartesian($a)
{
if ($a)
{
if($u=array_pop($a))
foreach(cartesian($a)as$p)
foreach($u as$v)
yield $p+[count($p)=>$v];
}
else
yield[];
}
Примечание: это не сохранить ключи, но его начало.
Это должно сделать (не проверено):
function acartesian($a)
{
if ($a)
{
$k=end(array_keys($a));
if($u=array_pop($a))
foreach(acartesian($a)as$p)
foreach($u as$v)
yield $p+[$k=>$v];
}
else
yield[];
}
другое решение:
function getAllVariations($input) {
$result = array();
$cnt = array_product(array_map('count', $input));
$step = 1;
foreach ($input as $key=>$array) {
for ($i=0; $i<$cnt; $i++) {
foreach ($array as $value) {
for ($k=0; $k<$step; $k++) {
$result[$i+$k][$key] = $value;
}
$i += $step;
}
$i--;
}
$step = $step * count($array);
}
return $result;
}
использование:
$input = array(
'arm' => array('A', 'B', 'C'),
'gender' => array('Female', 'Male'),
'location' => array('Vancouver', 'Calgary'),
'name' => array('Rio', 'Mark')
);
echo "<pre>";
var_dump(getAllVariations($input));
в PHP 7 @Serg ответ может быть сокращен до:
function cartesian(array $input)
{
$result = [[]];
foreach ($input as $key => $values) {
$append = [];
foreach ($values as $value) {
foreach ($result as $data) {
$append[] = $data + [$key => $value];
}
}
$result = $append;
}
return $result;
}
я быстро скорректировал ваш код немного, моя попытка грубая, я думаю, но посмотреть, работает ли это так, как вы хотите:
$result = array();
$nm = '';
foreach ($map as $name => $a) {
if (empty($result)) {
$result = $a;
$nm = $name;
continue;
}
$res = array();
foreach ($result as $r) {
foreach ($a as $v) {
$myr = $r;
$myv = $v;
if(!is_array($r)) $myr = array($nm => $r);
if(!is_array($v)) $myv = array($name => $v);
$res[] = array_merge($myr, $myv);
}
}
$result = $res;
}
echo "<pre>";
print_r($result);
почему бы не использовать базу данных для этого?
это легко в MySQL..
table arm
id integer primary key
label char
table gender
id integer primary key
gender enum('male','female')
table location
id integer primary key
city varchar(255)
затем сделайте запрос
$query = mysql_query("
SELECT a.label, g.gender, l.city
FROM arm a
CROSS JOIN gender g
CROSS JOIN location l
ORDER BY a.id
") or die("Could not execute query");
while($row = mysql_fetch_array($query) )
{
....
}
и прочитайте это:
если потребление памяти важно или вам не нужны все комбинации в конце концов, вы можете использовать итератор для создания одной комбинации за раз. Если вам нужны все комбинации, которые вы можете использовать iterator_to_array
.
function cartezianIterator($inputArray)
{
$maximumPosition = array_map('count', $inputArray);
$position = array_pad([], count($inputArray), 0);
while (false !== ($item = buildItemAtPosition($inputArray, $position))) {
yield $item;
$position = incrementPosition($position, $maximumPosition);
}
}
function buildItemAtPosition($inputArray, $positions)
{
if ($positions[0] >= count($inputArray[0])) {
return false;
}
$item = [];
foreach ($inputArray as $rowIndex => $row) {
$position = $positions[$rowIndex];
$item[] = $row[$position];
}
return $item;
}
function incrementPosition($position, $maximumPosition)
{
$digitToIncrement = count($position) - 1;
do {
$position[$digitToIncrement]++;
if ($position[$digitToIncrement] < $maximumPosition[$digitToIncrement] || 0 === $digitToIncrement) {
//no overflow
break;
}
//overflow, reset to zero and increment parent digit
$position[$digitToIncrement] = 0;
$digitToIncrement--;
} while ($digitToIncrement >= 0);
return $position;
}
затем, чтобы получить одно решение за раз, вы можете использовать foreach
или next
, например:
$iterator = cartezianIterator($inputArray);
//of course, you need to do something with the result...
$combination = next($iterator);
$combination = next($iterator);
$combination = next($iterator);
$combination = next($iterator);
$combination = next($iterator);
$combination = next($iterator);
это решение очень быстро, если вам нужно всего несколько комбинаций.
Кроме того, потребление памяти очень низкое (он использует flat array
сохранить некоторые integers
).
Примечание: рекурсивные функции не используются.
один алгоритм должен расширять на каждом шаге предыдущие результаты с текущими элементами шага:
function cartezian1($inputArray)
{
$results = [];
foreach ($inputArray as $group) {
$results = expandItems($results, $group);
}
return $results;
}
function expandItems($sourceItems, $tails)
{
$result = [];
if (empty($sourceItems)) {
foreach ($tails as $tail) {
$result[] = [$tail];
}
return $result;
}
foreach ($sourceItems as $sourceItem) {
foreach ($tails as $tail) {
$result[] = array_merge($sourceItem, [$tail]);
}
}
return $result;
}
Это решение использует память для хранения всех комбинаций, а затем возвращает их все сразу. Это быстро, но требует много памяти. Кроме того, рекурсивные функции не используются.