php-算法 找出一組數中比每個數大的右邊第一個數

<?php

$args = [1, 9, 10, 3, 2, 7, 5, 2, 6, 5, 5, 6];
//預計輸出 9，10，-1，7，7，-1，6，6，-1, 6, 6, -1
$res = [];
$stack = [];
$j = $r = 0;
for ($i = 0; $i < count($args); $i++){
    while ($j > 0 && $args[$i] > $args[$stack[$j-1]]) {
            $res[$stack[$j-1]] = $args[$i];
            $j--;
    }
    $stack[$j++] = $i;
    
}
if ($j > 0) {
    for ($i = 0; $i < $j; $i++) {
        $res[$stack[$i]] = -1;
    }
}
ksort($res);
print_r($res);