HDU 5722 Jewelry

矩形面积并。

需要转化一下思路：记录每一个位置的数以及位置。

对数字进行从小到大排序，数字一样的按位置从小到大排。

这样，一样的数就在一起了。连续的相同的x个数就可以构成很多解，这些解对应于二维平面上某矩形内整点个数（包括边界）。

这样就处理出很多很多的矩形，因为重复只能计算一次，所以可以采用矩形面积并计算。

PS：计算整点个数的话，可以讲每一个矩形的左下角左边-1 -1，这样就直接当面积来算就可以了。

#include <cstdio>
#include <cstring>
#include <cctype>
#include <algorithm>
using namespace std;
#define lson l , m , rt << 1
#define rson m + 1 , r , rt << 1 | 1

const long long maxn =  + ;
struct XX
{
    long long num;
    long long pos;
}s[maxn];
long long cnt[maxn << ];
long long sum[maxn << ];
long long X[maxn];
struct Seg {
    long long h, l, r;
    long long s;
    Seg(){}
    Seg(long long a, long long b, long long c, long long d) : l(a), r(b), h(c), s(d) {}
    bool operator < (const Seg &cmp) const {
        return h < cmp.h;
    }
}ss[maxn];
void PushUp(long long rt, long long l, long long r) {
    if (cnt[rt]) sum[rt] = X[r + ] - X[l];
    else if (l == r) sum[rt] = ;
    else sum[rt] = sum[rt << ] + sum[rt <<  | ];
}
void update(long long L, long long R, long long c, long long l, long long r, long long rt) {
    if (L <= l && r <= R) {
        cnt[rt] += c;
        PushUp(rt, l, r);
        return;
    }
    long long m = (l + r) >> ;
    if (L <= m) update(L, R, c, lson);
    if (m < R) update(L, R, c, rson);
    PushUp(rt, l, r);
}
long long Bin(long long key, long long n, long long X[]) {//离散化
    long long l = , r = n - ;
    while (l <= r) {
        long long m = (l + r) >> ;
        if (X[m] == key) return m;
        if (X[m] < key) l = m + ;
        else r = m - ;
    }
    return -;
}
bool cmp(const XX&a, const XX&b)
{
    if (a.num == b.num) return a.pos<b.pos;
    return a.num<b.num;
}
int main() {
    long long T, sz, h;
    scanf("%lld", &T);
    while (T--){

        scanf("%lld%lld", &sz, &h);
        for (long long i = ; i <= sz; i++)
        {
            scanf("%lld", &s[i].num);
            s[i].pos = i;
        }

        sort(s + , s +  + sz, cmp);

        long long m = ;

        for (long long i = ; i <= sz; i++)
        {
            if (i + h - >sz) break;
            if (s[i].num != s[i + h - ].num) continue;

            long long Lmin, Lmax, Rmin, Rmax;

            if (i ==  || s[i - ].num != s[i].num) Lmin = ;
            else Lmin = s[i - ].pos + ;
            Lmax = s[i].pos;

            Rmin = s[i + h - ].pos;
            if (i + h - == sz || s[i + h].num != s[i + h - ].num) Rmax = sz;
            else Rmax = s[i + h].pos - ;

            long long x1 = Lmin - , y1 = Rmin - , x2 = Lmax, y2 = Rmax;

            long long a = x1, b = y2, c = x2, d = y1;
            X[m] = a;
            ss[m++] = Seg(a, c, b, );
            X[m] = c;
            ss[m++] = Seg(a, c, d, -);
        }

        sort(X, X + m);
        sort(ss, ss + m);
        long long k = ;
        for (long long i = ; i < m; i++) {
            if (X[i] != X[i - ]) X[k++] = X[i];
        }
        memset(cnt, , sizeof(cnt));
        memset(sum, , sizeof(sum));
        long long ret = ;
        for (long long i = ; i < m - ; i++) {
            long long l = Bin(ss[i].l, k, X);
            long long r = Bin(ss[i].r, k, X) - ;
            if (l <= r) update(l, r, ss[i].s, , k - , );
            ret += sum[] * (ss[i + ].h - ss[i].h);
        }
        printf("%lld\n", ret);
    }
    return ;
}