天真的解决方案
您可以使用array_filter
根据性别创建两个组。然后将组压缩成对使用array_map
并运行对array_reduce
以展平结构:
$males = array_filter($students, function ($e) {
return $e["gender"] === "male";
});
$females = array_filter($students, function ($e) {
return $e["gender"] === "female";
});
$zipped = array_map(null, $males, $females);
$result = array_reduce($zipped, function ($a, $e) {
if ($e[0]) $a[] = $e[0];
if ($e[1]) $a[] = $e[1];
return $a;
}, []);
时间复杂度为 O(n)。
减少开销
如果第一个解决方案的开销太大,请考虑消除函数调用。两遍仍然是 O(n),但分支预测应该处理合并循环中性别之间数量不平衡的情况:
foreach ($students as $student) {
if ($student["gender"] === "male") {
$males[]= $student;
}
else {
$females[]= $student;
}
}
$male_count = count($males);
$female_count = count($females);
for ($i = 0, $j = 0; $i < $male_count || $j < $female_count;) {
if ($i < count($males)) {
$result[]= $males[$i++];
}
if ($j < count($females)) {
$result[]= $females[$j++];
}
}
概括
上面的代码假设了两件事:(1)"male"
应该始终是第一个,即使它产生了次优的交错(根据OP 的规范)和(2)只"gender"
存在两个值。
第一个问题可以通过修改上面的代码片段来解决,在压缩阶段交换数组顺序,优先选择最长的数组。
第二个问题可以通过array_reduce
为目标键的每个唯一值创建数组元素的分组来解决,然后删除硬编码的值,以便对这些按频率降序排序的组进行迭代(可以添加打破平局的逻辑)。
以下代码的时间复杂度为 O(n + k*log(k)),其中k
是唯一值的数量。最坏的情况是,所有条目都是完全或几乎唯一的,在这种情况下,由于多余的排序,我们有一个 O(n log(n)) 解决方案,但如果k
是常数,则它是 O(n),就像在 OP 的情况下一样。
请注意,PHP 排序例程不稳定,因此您需要将数组打包和解包为索引/元素对,或者使用除索引之外的自定义平局策略。
<?php
function interleave_values($arr, $key) {
$unique_values = array_unique(array_column($arr, $key));
$buckets = array_reduce($arr, function ($a, $e) use ($key) {
$a[$e[$key]][] = $e;
return $a;
}, []);
rsort($buckets);
$zipped = array_map(null, ...$buckets);
return array_reduce($zipped, function ($a, $e) {
foreach ($e as $f) {
if (!$f) break;
$a[] = $f;
}
return $a;
}, []);
}
$test = [
["k" => 1],
["k" => 2],
["k" => 1],
["k" => 3],
["k" => 3],
["k" => 1],
["k" => 2],
["k" => 2],
["k" => 2],
];
var_export(interleave_values($test, "k"));
输出:
array (
0 =>
array (
'k' => 2,
),
1 =>
array (
'k' => 1,
),
2 =>
array (
'k' => 3,
),
3 =>
array (
'k' => 2,
),
4 =>
array (
'k' => 1,
),
5 =>
array (
'k' => 3,
),
6 =>
array (
'k' => 2,
),
7 =>
array (
'k' => 1,
),
8 =>
array (
'k' => 2,
),
)