描述
题解
按照我的思路,这里我们首先通过第二个方法恢复到原序列,接着呢,就是按照一方法遍历一遍即可。
这里我们很容易想到,我们需要用到递归来恢复序列,那么怎么递归呢?仔细观察可以发现,方法二表示这个 pop 元素被压上了几个,那么我们可以将序列划分为两部分进行递归,一个是在该元素下的,一个是在该元素上的,所以呢,当我们处理过前者后,需要对该元素进行一次 push 操作,当我们处理过后者后,需要对该元素进行一次 pop 操作,这里我们可以使用 vector 来记录 push 和 pop 的过程,当然这部分用数组也是完全可以的。
另外,注意输入量有些感人,所以呢,注意输入输出外挂哦~~~
代码
#include <stdio.h>
#include <vector>
using namespace std;
const int MAXN = 1e6 + 5;
int sta[MAXN];
vector<int> vi;
// 1 push 0 pop
void solve(int l, int r)
{
if (l > r)
{
return ;
}
solve(l, r - sta[r] - 1);
vi.push_back(1);
solve(r - sta[r], r - 1);
vi.push_back(0);
}
template <class T>
inline void scan_d(T &ret)
{
char c;
ret = 0;
while ((c = getchar()) < '0' || c > '9');
while (c >= '0' && c <= '9')
{
ret = ret * 10 + (c - '0'), c = getchar();
}
}
inline void out(int x)
{
if (x > 9)
{
out(x / 10);
}
putchar(x % 10 + '0');
}
int main()
{
int n;
scan_d(n);
for (int i = 1; i <= n; i++)
{
scan_d(sta[i]);
}
solve(1, n);
int res = 0;
bool flag = true;
for (int i = 0; i < vi.size(); i++)
{
if (vi[i] == 1)
{
res++;
}
else
{
if (flag)
{
flag = false;
}
else
{
putchar(32);
}
out(res);
}
}
return 0;
}