hdu 4638 Group 线段树/数状数组+离线

#include <iostream>
#include <stdio.h>
#include <algorithm>
using namespace std;
#define maxn 100001
struct node{
    int l,r,w;//w记录(l,r)区间内的组数
}e[maxn*4];
struct node2{
    int l,r,id;
}f[maxn];
int cmp(node2 a,node2 b)
{
    return a.r<b.r;
}
int a[maxn],pos[maxn],ans[maxn];
void build(int a,int b,int c)
{
    e[c].l=a;
    e[c].r=b;
    e[c].w=0;
    if(a==b)
        return;
    int mid=(a+b)/2;
    build(a,mid,2*c);
    build(mid+1,b,2*c+1);
}
void update(int a,int b,int c,int val)
{
    if(e[c].l==a&&e[c].r==b)
    {
        e[c].w+=val;
        return;
    }
    int mid=(e[c].l+e[c].r)/2;
    if(b<=mid)
        update(a,b,2*c,val);
    else
        update(a,b,2*c+1,val);
    e[c].w=e[2*c].w+e[2*c+1].w;
}
int query(int a,int b,int c)
{
    if(e[c].l==a&&e[c].r==b)
        return e[c].w;
    int mid=(e[c].l+e[c].r)/2;
    if(b<=mid)
        return query(a,b,2*c);
    else if(a>mid)
        return query(a,b,2*c+1);
    else
        return query(a,mid,2*c)+query(mid+1,b,2*c+1);
}
int main()
{
    int T,n,m;
    scanf("%d",&T);
    while(T--)
    {
        int i,j,k;
        scanf("%d%d",&n,&m);
        for(i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            pos[a[i]]=i;
        }
        for(i=1;i<=m;i++)
        {
            scanf("%d%d",&f[i].l,&f[i].r);
            f[i].id=i;
        }
        sort(f+1,f+m+1,cmp);//将查找区间以右右区间递增排序，方便在更改线段树的时候查找，不会漏掉
        build(1,n,1);
        j=1;
        for(i=1;i<=n;i++)
        {
            update(i,i,1,1);//每次新进的一位数，假设独立，所有已i结尾的区间都+1
            //每次删除掉之前加入的与a[i]相邻的数，因为它们在一组里
            if(a[i]<n&&pos[a[i]+1]<i)
                update(pos[a[i]+1],pos[a[i]+1],1,-1);
            if(a[i]>1&&pos[a[i]-1]<i)
                update(pos[a[i]-1],pos[a[i]-1],1,-1);
            while(j<=m&&f[j].r==i)
            {
                ans[f[j].id]=query(f[j].l,f[j].r,1);
                j++;
            }
        }
        for(i=1;i<=m;i++)
            printf("%d\n",ans[i]);
    }
}
/*
题意：找到区间能，有多少组连续数字串
注意不要用cin和cout，会超时的。。
*/

//下面是树状数组的代码

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define maxn 100001
#define lowbit(x) ((x)&(-x))
int c[maxn],n,m,a[maxn],pos[maxn],ans[maxn];
struct node{
    int l,r,id;
}f[maxn];
int cmp(node a,node b)
{
    return a.r<b.r;
}
void add(int i,int val)
{
    while(i<=n)
    {
        c[i]+=val;
        i+=lowbit(i);
    }
}
int sum(int i)
{
    int s=0;
    while(i>=1)
    {
        s+=c[i];
        i-=lowbit(i);
    }
    return s;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int i,j,k;
        scanf("%d%d",&n,&m);
        for(i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            pos[a[i]]=i;
        }
        for(i=1;i<=m;i++)
        {
            scanf("%d%d",&f[i].l,&f[i].r);
            f[i].id=i;
        }
        sort(f+1,f+m+1,cmp);
        memset(c,0,sizeof(c));
        j=1;
        for(i=1;i<=n;i++)
        {
            add(i,1);
            if(a[i]<n&&pos[a[i]+1]<i)
                add(pos[a[i]+1],-1);
            if(a[i]>1&&pos[a[i]-1]<i)
                add(pos[a[i]-1],-1);
            while(j<=m&&f[j].r==i)
            {
                ans[f[j].id]=sum(f[j].r)-sum(f[j].l-1);
                j++;
            }
        }
        for(i=1;i<=m;i++)
            printf("%d\n",ans[i]);
    }
}

//比较水的线段树，可我就是写不出来，55555555