洛谷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

我写这道题就是为了学单调队列。。。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
#define N 1000003
struct node{
    int x,y;
}v[N];
int a[N],mx[N],mn[N],m,n; 
void getmin()
{
    int i,head = 1,tail = 0;
    for(i = 1;i <= m - 1;i++)
    {
        while(head <= tail && v[tail].x >= a[i])tail--;
        v[++tail].x = a[i];//tail要增加呀 
        v[tail].y = i;
    }
    for(i = m - 1;i <= n;i++)
    {
        while(head <= tail && v[tail].x >= a[i])tail--;
//开一个数组存最小值，如果有一个数位置比它靠前同时又比它大，那就把那个数踢出数组换成这个数
        v[++tail].x = a[i];
        v[tail].y = i;	
        while(v[head].y < i - m + 1) head++;
        mn[i - m + 1] = v[head].x;	
    } 
}
void getmax()
{
    int i,head = 1,tail = 0;
    for(i = 1;i <= m - 1;i++)
    {
        while(head <= tail && v[tail].x <= a[i])tail--;
        v[++tail].x = a[i];//tail要增加呀 
        v[tail].y = i;
    }
    for(i = m - 1;i <= n;i++)
    {
        while(head <= tail && v[tail].x <= a[i])tail--;
        v[++tail].x = a[i];
        v[tail].y = i;	
        while(v[head].y < i - m + 1) head++;//当首尾之间距离大于m时，我们就应该行动了
        mx[i - m + 1] = v[head].x;	
    } 
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i = 1;i <= n;i++)
    {
        scanf("%d",&a[i]);
    }
    getmin();
    getmax();
    for(int i = 1;i <= n - m + 1;i++)
    {
        if(i != 1)printf(" "); 
        printf("%d",mn[i]);
    }
    printf("\n");
    for(int i = 1;i <= n - m + 1;i++)
    {
        if(i != 1)printf(" ");
        printf("%d",mx[i]); 
    }
    return 0;
}