【洛谷】【线段树】P1886 滑动窗口

【题目描述：】

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

【输入格式：】

输入一共有两行，第一行为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

 

【算法分析:】 

线段树使用结构体同时维护区间最小值和最大值

没有修改只有建树和查询操作，更优的方法是使用st表做RMQ

板子题.

 

【代码：】

//滑动窗口
#include<iostream>
#include<cstdio>
using namespace std;

const int MAXN = 1e6 + 1;

int n, k;
int a[MAXN];
struct Segment {
    int maxn, minn;
}t[MAXN << 2];

void Build(int o, int l, int r) {
    if(l == r) t[o].maxn = t[o].minn = a[l];
    else {
        int mid = (l + r) >> 1;
        Build(o << 1, l , mid);
        Build(o << 1|1, mid + 1, r);
        t[o].maxn = max(t[o << 1].maxn, t[o << 1|1].maxn);
        t[o].minn = min(t[o << 1].minn, t[o << 1|1].minn);
    }
}
int max_ans, min_ans;
void Query(int o, int l, int r, int ql, int qr) {
    if(ql <= l && r <= qr) {
        max_ans = max(max_ans, t[o].maxn);
        min_ans = min(min_ans, t[o].minn);
    }
    else {
        int mid = (l + r) >> 1;
        if(ql <= mid) Query(o << 1, l, mid, ql, qr);
        if(qr > mid) Query(o << 1|1, mid + 1, r, ql, qr);
    }
}

int ans1[MAXN], ans2[MAXN];
int main() {
    scanf("%d%d", &n, &k);
    for(int i = 1; i <= n; ++i)
        scanf("%d", &a[i]);
    Build(1, 1, n);
    for(int i = 1; i + k - 1 <= n; ++i) {
        max_ans = -2e9, min_ans = 2e9;
        Query(1, 1, n, i, i + k - 1);
        ans1[i] = max_ans, ans2[i] = min_ans;
    }
    for(int i = 1; i <= n - k + 1; ++i) printf("%d ", ans2[i]);
    putchar('\n');
    for(int i = 1; i <= n - k + 1; ++i) printf("%d ", ans1[i]);
}