Algoogle

Algorithm for Programming Contest

POI XVII Pilots

Category: POI Tag: inchworm-method, slide-min-max

Pilots

問題概要


長さnの数列aの連続する部分列で最大値と最小値の差がt以下になるものの最大の長さを求めよ

解法


Segment Treeでやるとlogがついて計算量的に厳しい.
スライド最小値と最大値をしゃくとりしながらやる.
最大値-最小値がt以下ならまだ伸ばせるかもしれないので1つ伸ばす.
そうでないなら1つ縮める.

コード


(pilots.cpp) download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
#include <bits/stdc++.h>
using namespace std;

int t, n, a[1<<22];

void right(int &r, deque<int> &lb, deque<int> &ub)
{
        while(!lb.empty() and a[lb.back()]>=a[r]) lb.pop_back();
        while(!ub.empty() and a[ub.back()]<=a[r]) ub.pop_back();
        lb.push_back(r); ub.push_back(r);
        r++;
}

void left(int &l, deque<int> &lb, deque<int> &ub)
{
        if(lb.front() == l) lb.pop_front();
        if(ub.front() == l) ub.pop_front();
        l++;
}

int solve()
{
        deque<int> lb, ub;
        int l = 0, r = 0, ans = 1;
        while(l < n) {
                if(l == r) right(r,lb,ub);
                else if(a[ub.front()]-a[lb.front()] > t) left(l,lb,ub);
                else {
                        ans = max(ans, r-l);
                        if(r == n) break;
                        right(r,lb,ub);
                }
        }
        return ans;
}

void input()
{
        scanf("%d%d", &t, &n);
        for (int i = 0; i < n; i++) scanf("%d", a+i);
}

int main()
{
        input();
        printf("%d\n", solve());
        return 0;
}

Comments