算法--用栈操作构建数组

90 阅读2分钟

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

题目

leetcode 1441. 用栈操作构建数组 难度:中等

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

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

"Push":从 list 中读取一个新元素, 并将其推入数组中。

"Pop":删除数组中的最后一个元素。

如果目标数组构建完成,就停止读取更多元素。 题目数据保证目标数组严格递增,并且只包含 1 到 n 之间的数字。

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

 

示例 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 严格递增

题解

根据题意进行模拟即可:每次我们将当前处理到 i 压入栈中(往答案添加一个 Push),然后判断当前处理到的 i 是否最新的栈顶元素 target[j]target[j] 是否相同,若不相同则丢弃元素(往答案添加一个 Pop),若存在则将指针 j 后移,直到构建出目标答案。

/**
 * @param {number[][]} matrix
 * @return {number[]}
 */
var buildArray = function(target, n) {
    const res = [];
    let prev = 0;
    for (const number of target) {
        for (let i = 0; i < number - prev - 1; i++) {
            res.push("Push");
            res.push("Pop");
        }
        res.push("Push");
        prev = number;
    }
    return res;
};

代码详解

list集合是严格递增的,即: list = { 1 , 2 , 3 ..., n },而且target数组也是严格递增的,因为我们最终的结果是要获取拼装成target的“动作”(“Push” / “Pop”)集合,所以,我们只需要遍历targe数组,使其与递增的num进行对比,因为num的初始值为1,每次对比num都执行加1操作,所以在最终遍历之后,它的值就是1,2,3……递增的,可以代表题目中所描述的集合list。

当target[i]等于num的时候,则向result中插入“Push”字符串;当target[i]不等于num的时候,则向result中插入“Push”和“Pop”字符串;依次类推,当遍历完target数组后,将最终的结果result返回即可。