hdu 4638 Group

链接：http://acm.hdu.edu.cn/showproblem.php?pid=4638

大意：询问区间内有多少段连续的数，如3 1 2 5 4，区间[2,4]内有2段数，1～2和5

解法：考察当前插入一个数x会发生什么变化：如果x+1和x-1都已出现，那么x的插入会使得他们3个数连成一段，段的个数减少1；如果两者均没有出现，那么段的个数增加1；如果只出现其一，则段的个数不发生变化；那么，我们可以先将所有的数，按照这样的规律所取得的值插入到一颗线段树中，整个区间段的个数即为区间和；但是，询问不一定是整个区间，所以还需要删除前面的一些数，进一步，我们将所有询问读入后，按照左端点排序，对于每一个询问，删除左边的数，使得区间的左端点和询问区间的左端点相等；删除数会产生一些变化，例如3 1 2 5 4，一开始其值为1 1 -1 1 -1，删除3后，会对2，4的值产生影响，需要加一，表示段数增加，变为1 0 1 0，所以删除数x的时候，需要对x+1和x-1的值做加1的更新；每一个询问的答案就是query(q.x, q.y, 1, n, 1)

代码如下：

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

const int M = 111111;
int n, m, sum[M<<2], ans[M], pos[M], a[M], id[M];
bool use[M];

void pushup(int rt) { sum[rt] = sum[rt<<1] + sum[rt<<1|1]; }

void update(int p, int x, int l, int r, int rt)
{
    if (l == r) { sum[rt] += x; return ; }
    int m = (l+r)/2;
    if (p <= m) update(p, x, lson);
    else update(p, x, rson);
    pushup(rt);
}

int query(int L, int R, int l, int r, int rt)
{
    if (L <= l && r <= R) return sum[rt];
    int m = (l+r)/2, ret = 0;
    if (L <= m) ret += query(L, R, lson);
    if (R > m) ret += query(L, R, rson);
    return ret;
}

struct Q
{
    int x, y;
};
Q q[M];

int cmp(const int i, const int j) { return q[i].x < q[j].x; }

int main()
{
    int T; scanf("%d", &T);
    while ( T -- )
    {
        int n, m; scanf("%d%d", &n, &m);
        for (int i=1; i<=n; i++)
        {
            scanf("%d", &a[i]);
            pos[a[i]] = i;
        }
        memset(sum, 0, sizeof(sum));
        memset(use, 0, sizeof(use));
        for (int i=1; i<=n; i++)
        {
            if (use[a[i]-1] && use[a[i]+1]) update(i, -1, 1, n, 1);
            else if (!use[a[i]-1] && !use[a[i]+1]) update(i, 1, 1, n, 1);
            use[a[i]] = 1;
        }
        for (int i=0; i<m; i++) 
        {
            scanf("%d%d", &q[i].x, &q[i].y);
            id[i] = i;
        }
        sort(id, id+m, cmp);
        int s = 1;
        for (int i=0; i<m; i++)
        {
            int k = id[i];
            while (s <= n && q[k].x > s) //删除左边的数，使得区间的左端点与询问区间左端点一致 
            {
                if (pos[a[s]+1] > s) update(pos[a[s]+1], 1, 1, n, 1);
                if (pos[a[s]-1] > s) update(pos[a[s]-1], 1, 1, n, 1);
                s ++;
            }
            ans[k] = query(q[k].x, q[k].y, 1, n, 1);
        }
        for (int i=0; i<m; i++) printf("%d\n", ans[i]);
    }
    //system("pause");
    return 0;
}