I need to get all the nodes at a certain level in a full binary tree from either the left or right subtree. I currently retrieve the binary tree from the DB as an array, for exa
I upvoted BeetleJuice's answer for using the "Shift Left" bitwise operator <<
-- it is the perfect building block for this task.
This is as clean as I can make my coding attempt:
Code: (Demo)
function getForkinValues($array,$level,$side='left'){ // left side fork is default
if($level==1) return current($array);
$length=$level>2?1<<($level-2):1; // number of elements to return
$index=(1<<$level-1)-1; // first index to return
if($side=='right') $index+=$length; // shift to correct index for 'right'
if(!isset($array[$index+$length-1])) return 'Error: Insufficient Array Length';
return array_slice($array,$index,$length);
}
$array=['A','B','C','D','E','F','G'];
var_export(getForkinValues($array,3,'right'));
// Adjust depth as needed
$depth = 3;
// using bit arithmetic. '<< 3' means multiply by 2 three times.
$start = 1 << $depth-1; // 1 * (2 * 2) because depth is 3
$end = (1 << $depth) -1; // 1 * (2 * 2 * 2) - 1
// if depth=3, this is [4,5,6,7]
$fullLevel = range($start, $end);
print_r($fullLevel);
if($depth > 1):
$leftBranch = array_slice($fullLevel,0,count($fullLevel)/2);
$rightBranch = array_slice($fullLevel,count($fullLevel) / 2);
print_r($leftBranch); // [4,5]
print_r($rightBranch); // [6, 7]
endif;