leetcode 之 1441. 用栈操作构建数组

89 阅读1分钟

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

题目描述

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 <= target[i] <= 100
  • 1 <= n <= 100
  • target 是严格递增的

思路分析

  1. 1、初始化num为1
  2. for循环遍历数组target且每一次for循环后,num都递增1:
    1. 如果target中没有num,就进入while循环,res中就加入"Push"和"Pop",num也要递增1。如果target中仍然没有num,那么继续进入while循环,直到target中有num后跳出while循环
    2. 跳出while循环后,res中加入"Push",num递增1,i也递增1,再次进入for循环
    3. (3)遍历完target数组后跳出for循环
  3. 3、返回res结果

AC 代码

// 写法1
class Solution
{
public:
    vector<string> buildArray(vector<int> &target, int n)
    {
        vector<string> res;
        int num = 1;
        for (int i = 0; i < target.size(); ++i, num++)  // num++或++num均可
        {
            while (target[i] != num)
            /*
            易错点1:
            如果while写成了if
            Wrong Answer 18/49 cases passed (N/A)
            Testcase [2,3,4,5,8,9,10] 10
            Answer ["Push","Pop","Push","Push","Push","Push","Push","Pop","Push","Push","Pop","Push","Push"]
   Expected Answer ["Push","Pop","Push","Push","Push","Push","Push","Pop","Push","Pop","Push","Push","Push"]
            */
            {
                res.push_back("Push");
                res.push_back("Pop");
                num++;       // 易错点2:即使for循环中有num++,这里仍然也要写num++
            }
            res.push_back("Push");
            // for循环中有num++,这里不用再写num++
        }
        return res;
    }
};