The Preliminary Contest for ICPC Asia Nanjing 2019 F. Greedy Sequence_线段树

题目链接：F

题目大意：
给定一个序列a,长度为n,且其中的元素是1 - n的排列；我们需要构建n个序列记作s_i,
其中每一个s_i需要满足：
1.s_i[1] = i
2.s_i是一个递减序列
3.pos[s_i[j]] - pos[s_i[j - 1]] 的绝对值要小于等于k，即相邻的元素在序列a中的位置的差距不能超过k
最后输出每一个s_i序列中非0元素的个数

解题思路:
我们首先去找到s_i中第一个元素在a中的位置，然后再pos[i] - k - pos[i] + k这个区间里面去寻找小于i的最大值，找到的数值设为tmp,则count[i] = count[tmp] + 1，
注意：我们可以直接升序去寻找，这样对于每一个序列s_i我们只用去构建的线段树上去找一次就可以直接得到答案。
构建线段树：首先把a序列中数值1加到线段树中，然后以后每次算完一个s_i，就把a序列中数值 i 加到线段树中。

AC代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>

using namespace std;

const int maxn = 1e5 + 5;

int T, n, k, root, cnt;
int a[maxn], a_to[maxn], countt[maxn];
struct Node { int l, r, dat; }t[maxn * 4];

inline int build(void) {
	++ cnt;
	t[cnt].l = t[cnt].r = t[cnt].dat = 0;
	return cnt;
}

inline void insert(int &root, int l, int r, int ll, int rr, int x) {
	if(!root) root = build();
	if(l == r) { t[root].l = t[root].r = l, t[root].dat = x; return; }

	int mid = (l + r) >> 1;
	if(ll <= mid) insert(t[root].l, l, mid, ll, rr, x);
	if(mid < rr) insert(t[root].r, mid + 1, r, ll, rr, x); 
	if(t[root].l) t[root].dat = t[t[root].l].dat;
	if(t[root].r) t[root].dat = max(t[root].dat, t[t[root].r].dat);
}

inline int query(int root, int l, int r, int ll, int rr, int x) {
	if(!root) return -1;
	if(ll <= l && rr >= r) {
		if(t[root].dat < x) return t[root].dat;
		if(l == r) return -1;
		int mid = l + r >> 1;
		int tmp = query(t[root].l, l, mid, ll, rr, x);
		tmp = max(tmp, query(t[root].r, mid + 1, r, ll, rr, x));
		return tmp;
	}

	int mid = l + r >> 1, tmp1 = -1;
	if(ll <= mid) tmp1 = query(t[root].l, l, mid, ll, rr, x);
	if(rr > mid) tmp1 = max(tmp1, query(t[root].r, mid + 1, r, ll, rr, x));

	return tmp1;
}

int main(void) {
//	freopen("in.txt", "r", stdin);
	scanf("%d", &T);
	while(T --) {
		scanf("%d%d", &n, &k);
		for(int i = 1; i <= n; i ++) scanf("%d", &a[i]), a_to[a[i]] = i;
		
		countt[1] = 1;
		insert(root, 1, n, a_to[1], a_to[1], 1);
		for(int i = 2; i <= n; i ++) {
			int idx = a_to[i];
			int l = idx - k < 1 ? 1 : idx - k;
			int r = idx + k > n ? n : idx + k;
			int tmp = query(root, 1, n, l, r, i);
			int permt = 0, f = 0;
			
			if(tmp == -1) {
				countt[i] = 1;
			} else {
				while(tmp != -1) {
					permt ++;
					idx = a_to[tmp];
					if(countt[a[idx]]) {
						countt[i] = permt + countt[a[idx]];
						f = 1;
						break;
					}
					l = idx - k < 0 ? 0 : idx - k;
					r = idx + k > n ? n : idx + k;
					tmp = query(root, 1, n, l, r, i);
				}
				
				if(!f) countt[i] = permt;
			}
			
			insert(root, 1, n, a_to[i], a_to[i], i);
		}

		for(int i = 1; i <= n; i ++) 
			if(i == n) printf("%d\n", countt[i]);
			else printf("%d ", countt[i]);
		
		root = 0, cnt = 0;
		for(int i = 1; i <= n; i ++) countt[i] = 0;
	}

//	fclose(stdin);
	return 0;
}

总结：
在创建线段树的时候不要再一开始就把所有的节点加进去，我们可以算一个节点就加上一个节点；如果我们一开始就全部加进去会超时。