找剩余数组第一个比较当前元素大的下标位置

14 阅读1分钟
在学校中,N个小朋友站成一队, 第i个小朋友的身高为height[i],
第i个小朋友可以看到的第一个比自己身高更高的小朋友j,那么j是i的好朋友(要求j > i)。
请重新生成一个列表,对应位置的输出是每个小朋友的好朋友位置,如果没有看到好朋友,请在该位置用0代替。
小朋友人数范围是 [0, 40000]。

思路1:双层循环


let getFriendList = (heightArray) => {
    let res = [];
    for (let i = 0; i < heightArray.length; i++) {
        let curData = 0;
        for (let j = i+1; j< heightArray.length; j++) {
            if (heightArray[j] > heightArray[i]) {
                curData = j;
                break;
            }
        }
        res.push(curData);
    }
    return res;
}

console.log(getFriendList([1,2,5,2,1,3])); // [ 1, 2, 0, 5, 5, 0 ]

思路2:单调递减栈

let getFriendList2 = (heightArray) => {
    let stack = [];
    let map = {};

    for (let i = heightArray.length -1 ; i >= 0 ; i--) {

        while (stack.length > 0 && stack[stack.length - 1].value <= heightArray[i]) {

            if (stack[stack.length - 1].value <= heightArray[i]) {

                stack.pop();

            }

        }

        map[i] = stack[stack.length - 1] && stack[stack.length - 1].key || 0;
        stack.push({

            key: i,

            value: heightArray[i]

        });
    }
    return heightArray.map((item, index) => (map[index]));
}

console.log(getFriendList2([1,2,5,2,1,3])); // [ 1, 2, 0, 5, 5, 0 ]