PKU 2823 Sliding Window

利用单调队列的统计。类似的题目有PKU 3250

题目意思是要求给定数列中,一个大小固定的滑动窗口内的最值

维护一个单调队列,末端插入,弹出,首端删除即可。队首始终为最值。

需要注意的两个问题:

1. 队列的初始化

2. 队首弹出需判断

 

  1. #include 
  2. using namespace std;
  3. const int N = 1000010;
  4. int a[N];
  5. int deck1[N], deck2[N], head1, tail1, head2, tail2;
  6. int ans1[N], ans2[N], cnt;
  7. int n, m;
  8. inline void insert(int x)
  9. {
  10.     while(head1 <= tail1 && a[x] < a[deck1[tail1]]) tail1--;
  11.     deck1[++tail1] = x;
  12.     
  13.     while(head2 <= tail2 && a[x] > a[deck2[tail2]]) tail2--;
  14.     deck2[++tail2] = x;
  15. }
  16. int main()
  17. {
  18.     int i, j, k;
  19.     
  20.     scanf("%d%d",&n, &m);
  21.     for(i = 1; i <= n; i++) scanf("%d",&a[i]);
  22.     
  23.     head1 = tail1 = 1;
  24.     deck1[1] = 1; 
  25.     
  26.     head2 = tail2 = 1;
  27.     deck2[1] = 1; 
  28.     
  29.     for(i = 2; i <= m; i++)
  30.         insert(i);
  31.     
  32.     ans1[1] = a[deck1[1]];
  33.     ans2[1] = a[deck2[1]];
  34.     cnt = 1;
  35.     
  36.     for(; i <= n; i++)
  37.     {
  38.         insert(i);
  39.         if(deck1[head1] == i - m) head1++;
  40.         if(deck2[head2] == i - m) head2++;
  41.         ans1[++cnt] = a[deck1[head1]];
  42.         ans2[cnt] = a[deck2[head2]];
  43.     }
  44.     
  45.     for(i = 1; i <= cnt; i++)
  46.         printf("%d%c", ans1[i], i == cnt? '/n' : ' ');
  47.     for(i = 1; i <= cnt; i++)
  48.         printf("%d%c", ans2[i], i == cnt? '/n' : ' ');
  49.     
  50.     scanf("%d",&n);
  51.     return 0;
  52. }
  53.         
  54.         

 

 

 

你可能感兴趣的:(PKU 2823 Sliding Window)