题目地址:
https://leetcode.com/problems/build-an-array-with-stack-operations/
给定一个数组target,再给定一个数字 n n n,另外给一个栈,从 [ 1 , 2 , . . . , n ] [1,2,...,n] [1,2,...,n]中依次读数,可以将读到的数进栈或者从栈中出去一个数。求一个操作序列使得操作完后栈里的那些数字恰好等于target数组。
思路是直接模拟,遍历 [ 1 , 2 , . . . , n ] [1,2,...,n] [1,2,...,n],如果遍历到target中的数字,则意味着此时要push;若遍历到的数字不在target里,说明这个数字被先push后pop了。另外用一个变量记录用到target的哪个数字了,如果target数字用完了就直接退出循环。代码如下:
import java.util.ArrayList;
import java.util.List;
public class Solution {
public List<String> buildArray(int[] target, int n) {
List<String> res = new ArrayList<>();
if (target == null || target.length == 0) {
return res;
}
// 这个变量标记用到了target的数字的下标
int idx = 0;
for (int i = 1; i <= n; i++) {
if (target[idx] == i) {
res.add("Push");
idx++;
} else {
res.add("Push");
res.add("Pop");
}
if (idx == target.length) {
break;
}
}
return res;
}
}
时间复杂度 O ( min ( n , l ) ) O(\min(n,l)) O(min(n,l))( l l l是target的长度),空间 O ( 1 ) O(1) O(1)(不包括返回的答案)。