传送门:滑动窗口
题目描述
现在有一堆数字共N个数字(N<=10^6),以及一个大小为k的窗口。现在这个从左边开始向右滑动,每次滑动一个单位,求出每次滑动后窗口中的最大值和最小值。
例如:
The array is [1 3 -1 -3 5 3 6 7], and k = 3.
输入输出格式
输入格式:输入一共有两行,第一行为n,k。
第二行为n个数(<INT_MAX).
输出共两行,第一行为每次窗口滑动的最小值
第二行为每次窗口滑动的最大值
输入输出样例
说明
50%的数据,n<=10^5
100%的数据,n<=10^6
呃 。。 还是比较水的线段树,思路就是建两棵树,一棵保留最大值,一棵保留最小值,然后分别输出就可以了(第一次提交时 minn 取大了,maxx 取小了。。。扎心了!
ZKW 线段树代码
#include<iostream>
#include<cstdio>
using namespace std;
int n,m,M;
int a[5000000+100],t[5000000+100];
void buildmax()
{
for(int i=M-1;i>=1;i--)
{
a[i] = max(a[i<<1],a[i<<1|1]);
}
}
void buildmin()
{
for(int i=M-1;i>=1;i--)
t[i] = min(t[i<<1],t[i<<1|1]);
}
int querymin(int x,int y)
{
x--;y++;
x+=M;y+=M;
int minn = 1e9;
while(x^y^1)
{
if(~x&1) minn = min(minn,t[x^1]);
if(y&1) minn = min(minn,t[y^1]);
x>>=1;
y>>=1;
}
return minn;
}
int querymax(int x,int y)
{
x--;y++;
x+=M;y+=M;
int minn = -1e9;
while(x^y^1)
{
if(~x&1) minn = max(minn,a[x^1]);
if(y&1) minn = max(minn,a[y^1]);
x>>=1;
y>>=1;
}
return minn;
}
int main()
{
scanf("%d%d",&n,&m);
for(M=1;M<=n;M<<=1);
for(int i=M+1;i<=M+n;i++)
{
scanf("%d",&a[i]);
t[i] = a[i];
}
buildmax();
buildmin();
for(int i=1;i<=n-m+1;i++)
{
printf("%d ",querymin(i,i+m-1));
}
cout<<endl;
for(int i=1;i<=n-m+1;i++)
{
printf("%d ",querymax(i,i+m-1));
}
return 0;
}