在学校中,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 ]