HDU 5289 - Assignment(multiset + 尺取法)

题目：

http://acm.hdu.edu.cn/showproblem.php?pid=5289

题意：

求出序列中有多少个几个使得集合中的数是连续的且任意两个数差值小于k。

思路：

multiset 储存合格集合，每次新加入的数与最大值和最小值比较。

PS：set和map 的find 的复杂度是O（logn），vector 的find的复杂度是 O（n）。

AC.

#include <iostream>
#include <cstdio>
#include <cmath>
#include <set>

using namespace std;
int a[100005];
multiset<int> s;

int main()
{
    //freopen("in", "r", stdin);
    int T;
    scanf("%d", &T);
    int n, k;
    while(T--) {
        s.clear();
        scanf("%d %d", &n, &k);

        for(int i = 0; i < n; ++i) {
            scanf("%d", &a[i]);
        }
        if(k == 0) {
            printf("0\n");
            continue;
        }
        s.insert(a[0]);
        int l = 0, r = 1;

        int cnt = 1;
        long long ans = 0;
        int minn, maxn;
        while(1) {
            if(s.size()) {
                minn = *s.begin();
                maxn = *s.rbegin();

                if(abs(a[r]-minn) < k && abs(a[r]-maxn) < k) {
                    ans += (long long)s.size();
                    s.insert(a[r]);
                    r++;
                    if(r == n) break;
                }
                else {
                    if(s.size()) {
                        s.erase(s.find(a[l]));
                    }
                    l++;
                }
            }

            else {
                l = r;
                s.insert(a[r]);
                r++;
                if(r == n) break;
            }
        }
        printf("%I64d\n", ans+n);
    }
    return 0;
}