P1886 滑动窗口 -线段树基础练习

P1886 滑动窗口 -线段树基础练习

题目链接：https://www.luogu.org/problemnew/show/P1886

题目描述

现在有一堆数字共N个数字（N<=10^6），以及一个大小为k的窗口。现在这个从左边开始向右滑动，每次滑动一个单位，求出每次滑动后窗口中的最大值和最小值。

例如：

The array is [1 3 -1 -3 5 3 6 7], and k = 3.

输入输出格式
输入格式：
输入一共有两行，第一行为n,k。

第二行为n个数(<INT_MAX).

输出格式：
输出共两行，第一行为每次窗口滑动的最小值

第二行为每次窗口滑动的最大值

输入输出样例

输入样例#1：

8 3
1 3 -1 -3 5 3 6 7

输出样例#1：

-1 -3 -3 -3 3 3
3 3 5 5 6 7

说明

50%的数据，n<=10^5
100%的数据，n<=10^6

思路

一样是套模板，只不过在建树的时候要同时查找最大值和最小值，同时，查询时也要对最大值和最小值同时查找，这里用一个变量f即可区分，很简单，这题应该没有不会的吧

代码

#include <iostream>
#include <algorithm>
#define ll long long
#define inf 0x3f3f3f3f3f3f3f3f
#define maxn 10000005
#define _for(i, a) for(int i = 0; i < (a); i++)
#define _rep(i, a, b) for(int i = (a); i <= (b); i++)
using namespace std;
ll arr[maxn], segtree[maxn * 4][2];
void build(ll node, ll begin, ll end) {
	if (begin == end) {
		segtree[node][0] = segtree[node][1] = arr[begin];
		return;
	}
	ll mid = (begin + end) / 2;
	build(node * 2, begin, mid);
	build(node * 2 + 1, mid + 1, end);
	segtree[node][0] = min(segtree[node * 2][0], segtree[node * 2 + 1][0]);
	segtree[node][1] = max(segtree[node * 2][1], segtree[node * 2 + 1][1]);
}
ll query(ll node, ll begin, ll end, ll left, ll right, ll f) {
	if (begin == left && end == right) return segtree[node][f];
	ll mid = (begin + end) / 2;
	if (f == 0) {
		if (mid >= right) return query(node * 2, begin, mid, left, right, f);
		else if (mid < left) return query(node * 2 + 1, mid + 1, end, left, right, f);
		else return min(query(node * 2, begin, mid, left, mid, f), query(node * 2 + 1, mid + 1, end, mid + 1, right, f));
	}
	else {
		if (mid >= right) return query(node * 2, begin, mid, left, right, f);
		else if (mid < left) return query(node * 2 + 1, mid + 1, end, left, right, f);
		else return max(query(node * 2, begin, mid, left, mid, f), query(node * 2 + 1, mid + 1, end, mid + 1, right, f));
	}
}
int main() {
	//freopen("in.txt", "r", stdin);
	ios::sync_with_stdio(false);
	int n, k;
	cin >> n >> k; k--;
	_for(i, n) cin >> arr[i];
	build(1, 0, n - 1);
	_for(i, n - k) cout << query(1, 0, n - 1, i, i + k, 0) << " ";
	cout << "\n";
	_for(i, n - k) cout << query(1, 0, n - 1, i, i + k, 1) << " ";
	return 0;
}