2013 多校第四场 hdu 4638 Group

hdu 4638

题目：http://acm.hdu.edu.cn/showproblem.php?pid=4638

题目大意：给你n个数（1~n），任意顺序，然后id连续的人能被分成一个组，现在给你m个询问，每次查询一个区间，问你区间里的数最少能分成几个组。

思路：很明显的线段树题。先是离线，然后按照右端点排序，从左往右扫过去，对于当前位置i，数字为a[ i ]，先更新i这个位置的值+1，再看看a[ i ] + 1 和a[ i ] - 1，在不在i 的前面，如果是，那么就分别对那个位置-1。如果遇见正好是询问点，就更新好了直接询问就好了，区间和就是。

看到那么多队伍都出了，可自己就怎么也想不到怎么来建树，看看和上场比赛那道线段树题很像，离线，排序什么的都知道，可就是搞不出来，现在想想应该是自己对上场比赛那道线段树做法理解不够，才会写不出来的。比赛后问的同学才知道，其实就是模拟一下每个点更新的过程，找规律，每次都维护以当前位置为右端点的所有区间的值，有查询就查询，就这样，扫一遍就好了。

因为只要求和，可以用线段树，也可以是树状数组的，当然，其实都一样的。。。 = =

先是线段树，代码如下：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

const int MAXN = 111111 ;

#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1

struct Qu
{
    int l,r;
    int id;
    bool operator < (const Qu &tmp) const
    {
        return r<tmp.r;
    }
} qu[MAXN];

int a[MAXN];

int node[MAXN<<2];

void update(int l,int r,int rt,int x,int c)
{
    if(l==r)
    {
        node[rt] += c;
        return ;
    }
    int m = l+r>>1;
    if(x<=m)
        update(lson,x,c);
    else update(rson,x,c);
    node[rt] = node[rt<<1]+node[rt<<1|1];
}

int query(int l,int r,int rt,int a,int b)
{
    if(a<=l&&b>=r)
    {
        return node[rt];
    }
    int m = l+r>>1;
    int cnt1 = 0,cnt2 = 0 ;
    if(a<=m)
        cnt1  = query(lson,a,b);
    if(b>m) cnt2 = query(rson,a,b);
    return cnt1+cnt2;
}

void init()
{
    memset(node,0,sizeof(node));
}

int pos[MAXN];

int ans[MAXN];

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;
        }
        for(int i = 1;i<=m;i++)
        {
            scanf("%d%d",&qu[i].l,&qu[i].r);
            qu[i].id = i;
        }
        sort(qu+1,qu+m+1);
        int k = 1;
        init();
        for(int i = 1;i<=n;i++)
        {
            int x = a[i];
            update(1,n,1,i,1);
            if(x<n&&pos[x+1]<i)
                update(1,n,1,pos[x+1],-1);
            if(x>1&&pos[x-1]<i)
                update(1,n,1,pos[x-1],-1);
            //printf("i = %d,node[1] = %d\n",i,node[1]);
            while(qu[k].r==i&&k<=m)
            {
                ans[qu[k].id] = query(1,n,1,qu[k].l,qu[k].r);
                k++;
            }
        }
        for(int i = 1;i<=m;i++)
            printf("%d\n",ans[i]);
    }
    return 0;
}

然后顺便练练树状数组，又改了一下，写了一个树状数组的 = =，代码如下：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

const int MAXN = 111111 ;

#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1

struct Qu
{
    int l,r;
    int id;
    bool operator < (const Qu &tmp) const
    {
        return r<tmp.r;
    }
} qu[MAXN];

int a[MAXN];

int node[MAXN<<2];

int n;

int lowbit(int x)
{
    return x&(-x);
}

void update(int x,int c)
{
    while(x<=n)
    {
        node[x] += c;
        x += lowbit(x);
    }
}

int query(int a,int b)
{
    a--;
    int cnt1 = 0;
    while(b>0)
    {
        cnt1 += node[b];
        b -= lowbit(b);
    }

    int cnt2 = 0;
    while(a>0)
    {
        cnt2 += node[a];
        a -= lowbit(a);
    }
    return cnt1-cnt2;
}

void init()
{
    memset(node,0,sizeof(node));
}

int pos[MAXN];

int ans[MAXN];

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int m;
        scanf("%d%d",&n,&m);
        for(int i = 1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            pos[a[i]] = i;
        }
        for(int i = 1;i<=m;i++)
        {
            scanf("%d%d",&qu[i].l,&qu[i].r);
            qu[i].id = i;
        }
        sort(qu+1,qu+m+1);
        int k = 1;
        init();
        for(int i = 1;i<=n;i++)
        {
            int x = a[i];
            update(i,1);
            if(x<n&&pos[x+1]<i)
                update(pos[x+1],-1);
            if(x>1&&pos[x-1]<i)
                update(pos[x-1],-1);
            //printf("i = %d,node[1] = %d\n",i,node[1]);
            while(qu[k].r==i&&k<=m)
            {
                ans[qu[k].id] = query(qu[k].l,qu[k].r);
                k++;
            }
        }
        for(int i = 1;i<=m;i++)
            printf("%d\n",ans[i]);
    }
    return 0;
}