NYOJ 522 Interval （树状数组）

Interval
时间限制：2000 ms | 内存限制：65535 KB
难度：4
描述
There are n(1 <= n <= 100000) intervals [ai, bi] and m(1 <= m <= 100000) queries, -100000 <= ai <= bi <= 100000 are integers.
Each query contains an integer xi(-100000 <= x <= 100000). For each query, you should answer how many intervals covers xi.
输入
The first line of input is the number of test case.
For each test case,
two integers n m on the first line,
then n lines, each line contains two integers ai, bi;
then m lines, each line contains an integer xi.
输出
m lines, each line an integer, the number of intervals that covers xi.
样例输入
2
3 4
1 3
1 2
2 3
0
1
2
3
1 3
0 0
-1
0
1
样例输出
0
2
3
2
0
1
0
解题思路：本来不懂什么树状数组的我决心学一下什么是树状数组，花了一个晚上的时间搞懂了树状数组是怎么回事，就想找个题来练一下，就找到这个了。首先我们来了解一下什么是树状数组，所谓的树状数组类似线段树，平常我们会遇到一些对数组进行维护查询的操作，比较常见的如，修改某点的值、求某个区间的和，而这两种恰恰是树状数组的强项！当然，数据规模不大的时候，对于修改某点的值是非常容易的，复杂度是O(1)，但是对于求一个区间的和就要扫一遍了，复杂度是O(N)，如果实时的对数组进行M次修改或求和，最坏的情况下复杂度是O(M*N)，当规模增大后这是划不来的！而树状数组干同样的事复杂度却是O(M*lgN)，别小看这个lg，很大的数一lg就很小了，这个学过数学的都知道吧，不需要我说了。
下边我们看一下这两个图


这里的A数组是指原来的数组，C数组是我们的树状数组，首先我们先来理解lowbit（k）这个很重要的东西，lowbit(k)就是把k的二进制的高位1全部清空，只留下最低位的1,比如10的二进制是1010,则lowbit(k)=lowbit(1010)=0010(2进制)，介于这个lowbit在下面会经常用到，这里给一个非常方便的实现方式，比较普遍的方法lowbit(k)=k&-k，这是位运算，我们知道一个数加一个负号是把这个数的二进制取反+1，如-10的二进制就是-1010=0101+1=0110，然后用1010&0110，答案就是0010了！明白了求解lowbit的方法就可以了。C[k]表示从A[k]开始往左连续求lowbit（k）个数的和；
如：C1 = A1 lowbit(1)=1;
C2 = A1 + A2 lowbit(2)=2;
C3 = A3 lowbit(3)=1;
C4 = A1 + A2 + A3 + A4 lowbit(4)=4;
C5 = A5 lowbit(5)=1;
C6 = A5 + A6 lowbit(6)=2;
C7 = A7 lowbit(7)=1;
C8 = A1 + A2 + A3 + A4 + A5 + A6 + A7 + A8 lowbit(8)=8;
这个时候就可以很容易的看出这个树状数组其实就是一个二分的过程了，第二张图更加可以明显的看出。下面就来说数组更新的问题了，假如你更新了A[3]，则C[3],C[4],C[8]都需要更新，这个v时候lowbit（）又要发挥他的作用了，其实从0011——>0100——>1000的变化都是进行“去尾”操作，又是自己造的词–”，我来解释下，就是把尾部应该去掉的1都去掉转而换到更高位的1,记住每次变换都要有一个高位的1产生，所以0100是不能变换到0101的，因为没有新的高位1产生，这个变换过程恰好是可以借助我们的lowbit进行的，k +=lowbit(k)；好感觉到现在为止树状数组应该介绍的差不多了，尽管不能完全弄懂原理，至少他是怎么回事应该知道了吧。下边附上一个一般树状数组的模板函数

#include<iostream>
using namespace std;
int n,m,i,num[100001],t[200001],l,r;//num:原数组；t：树状数组 
int lowbit(int x)
{
    return x&(-x);
}
void change(int x,int p)//将第x个数加p 
{
    while(x<=n)
    {
        t[x]+=p;
        x+=lowbit(x);
    }
    return;
}
int sum(int k)//前k个数的和 
{
    int ans=0;
    while(k>0)
    {
        ans+=t[k];
        k-=lowbit(k);
    }
    return ans;
}
int ask(int l,int r)//求l-r区间和 
{
    return sum(r)-sum(l-1); 
}
int main()
{
    cin>>n>>m;
    for(i=1;i<=n;i++)
    {
        cin>>num[i];
        change(i,num[i]);
    }
    for(i=1;i<=m;i++)
    {
        cin>>l>>r;
        cout<<ask(l,r)<<endl;
    }
    return 0;
}

最后就是解决这个题了，解决这个题的时候用到了一个小技巧，就是在输入每个区间的时候，将这个区间的前端点+1，后端点的后一位-1；然后再对要查寻的那个数x求1~x的和，下面看源代码

#include <iostream>
#include <bits/stdc++.h>
using namespace std;
int t[200002];//num:原数组；t：树状数组
int lowbit(int x)
{
    return x&(-x);
}
void change(int x,int p)//将第x个数加p
{
    while(x<=200002)
    {
        t[x]+=p;
        x+=lowbit(x);
    }
    return;
}
int sum(int k)//前k个数的和
{
    int ans=0;
    while(k>0)
    {
        ans+=t[k];
        k-=lowbit(k);
    }
    return ans;
}
int main()
{
    int T;scanf("%d",&T);
    while(T--)
    {
        int n,m,a,b,x;
        memset(t,0,sizeof(t));
        scanf("%d %d",&n,&m);
        for(int i=0;i<n;i++)
        {
            scanf("%d %d",&a,&b);
            change(a+100001,1);//因为题目里说的有负数，为了方便计算将负数和0都转化为正数，这里的1和-1是指将头端点+1尾端点的下一个-1；
            change(b+100002,-1);
        }
        for(int i=0;i<m;i++)
        {
            scanf("%d",&x);
            printf("%d\n",sum(x+100001));//1~x求和
        }
    }
    return 0;
}