用栈操作构建数组

66 阅读2分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第19天,点击查看活动详情

1441. 用栈操作构建数组 - 力扣(LeetCode)

给你一个数组 target 和一个整数 n。每次迭代,需要从  list = { 1 , 2 , 3 ..., n } 中依次读取一个数字。

请使用下述操作来构建目标数组 target

  • "Push":从 list 中读取一个新元素, 并将其推入数组中。
  • "Pop":删除数组中的最后一个元素。
  • 如果目标数组构建完成,就停止读取更多元素。

题目数据保证目标数组严格递增,并且只包含 1n 之间的数字。

请返回构建目标数组所用的操作序列。如果存在多个可行方案,返回任一即可。

示例 1:

输入: target = [1,3], n = 3
输出: ["Push","Push","Pop","Push"]
解释:
读取 1 并自动推入数组 -> [1]
读取 2 并自动推入数组,然后删除它 -> [1]
读取 3 并自动推入数组 -> [1,3]

示例 2:

输入: target = [1,2,3], n = 3
输出: ["Push","Push","Push"]

示例 3:

输入: target = [1,2], n = 4
输出: ["Push","Push"]
解释: 只需要读取前 2 个数字就可以停止。

提示:

  • 1 <= target.length <= 100
  • 1 <= n <= 100
  • 1 <= target[i] <= n
  • target 严格递增

思路

本题很简单,因为构建目标数组只有两个操作PushPopPush是推入数组,Pop是删除最后一个元素,如果我们只考虑Push,会得到一个从1n的数组,如果数组中不存在i元素怎么办呢?只需要先推入(Push)再删除(Pop)即可。从1循环到target最大一个元素值,如果i存在target中只要Push操作,不存在,先PushPop,又因为target严格递增,可以使用双指针循环比较。

解题

/**
 * @param {number[]} target
 * @param {number} n
 * @return {string[]}
 */
var buildArray = function (target, n) {
  const end = target.at(-1);
  let res = [];
  for (let i = 1, j = 0; i <= end; i++) {
    if (i === target[j]) {
      res.push("Push");
      j++;
    } else {
      res.push("Push", "Pop");
    }
  }
  return res;
};