Tests for Haybales(思维/dfs/构造)

题目
题意：已知一个长度为$n（1<=n<=10^5）$的未知的非递减数组$a_i$。已知，对于数组中每个元素$a_i$，不小于$a_i+k$的最大下标为$x_i$，即$a_{x_i}<=a_i+k<a_{x_i+1}$。根据这一条条件，有$x_1<=x_2<=...<=x_n$。
现给定$x_1,x_2,...,x_n$，构造$k$和对应的$a$数组

参考
思路：对于位置$i$上的数，我们给它构造父节点$x_i+1$，表示位置$x_i+1$是第一个大于$a_{x_i}+k$的位置。根据$x_i$的定义，我们可以构造得到一棵树，根节点为$n+1$。
如样例2 2 4 5 6 6，对应的树为

我们令$k=n+1$
对于不同高度的元素，乘上对应深度的k，保证，高度大的元素，大于次高的元素+k。
对于相同高度的元素，为保证$a$数组的非递减，要保证下标大的元素，更大。因此，我们给它们一个偏移值，下标大的元素，大于下标小的元素。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 100010;

vector<int> ve[maxn];
int val[maxn], deep[maxn];
int n, x, value, k;
ll a[maxn];

void dfs(int u, int d) {
	deep[u] = d;
	val[u] = value--;// value为全局偏移值
	int sz = ve[u].size();
	// 倒序遍历，保证同层的情况下，遍历下标大的元素，
	// 使val[pos1] < val[pos2], pos1<pos2 && deep[pos1]==deep[pos2]
	for (int i = sz - 1; i >= 0; --i) {
		dfs(ve[u][i], d + 1);
	}
}
int main() {
	scanf("%d", &n);
	for (int i = 1; i <= n; ++i) {
		scanf("%d", &x);
		// 因为i递增，对于每个pos ve[pos]存储的下标值，是从小到大的
		ve[x+1].push_back(i);
	}
	value = n + 1;
	k = n + 1;
	dfs(n + 1, 0);
	printf("%d\n", k);
	for (int i = 1; i <= n; ++i) {
		a[i] = 1LL * (deep[1] - deep[i]) * k + val[i];
		printf("%lld\n", a[i]);
	}
}