function solution(heights) {
  return heights.map((h, i) => {
    while (heights[i] <= h) {
      i--;
    }
    return ++i;
  });
}