QDUOJ 37 帅气的HYC的珍珠(树状数组)

题目地址：点击打开链接

帅气的HYC经常早晨去锻炼(多么好的习惯～。有一天，他看到一路上的露珠，心里便产生了一个问题：一路上假如有N棵草，每颗草上可能会有露珠，或者没有露珠。连续的露珠会和为一体(>=2)，并变为珍珠。

比如第1棵草上有露珠，第2棵草也有露珠。那么就会形成一个珍珠。第1棵草上有露珠，第2棵草有露珠，第3棵也有露珠，那么也会形成一个珍珠。相反，如果第1颗草有露珠，第2棵草没有露珠，就不会形成珍珠。

现在给你HYC今天早晨一路上的看到的N棵草，以及上面的露珠情况，给你Q次询问，每次询问［L,R],你能快速的告诉HYC，这个区间能形成多少个珍珠吗？ACCEPTIT!

输入

第1行一个整数T，表示有T组测试数据。(T<=15)
第2行一个整数N， 表示有N棵草。(1<=N<=1000)
第3行有N个整数0或者1，中间用空格隔开， 1代表第i课草有露珠，0代表第i棵草没有露珠。
第4行是一个整数Q， 代表查询的次数。(1<=Q<=100000)
下面第（4+Q）行每行两个整数L， R。 代表查询的区间［L， R］；（1<=L<=R<=N）

输出

对于每组查询输出一行。

样例输入1  复制

1
4
1 1 0 1
3
1 2
2 3
1 4

样例输出1

1
0
1

思路：一串连续的露珠能形成一颗珍珠，一开始我们把这颗珍珠用树状数组记录在这串的最后一个位置上，每一次查

询通过树状数组查询这段区间的珍珠数，考虑下该区间的两端的露珠是否能形成珍珠。

#include
#include
#include
using namespace std;
const int maxn = 1e3+5;
int a[maxn], tree[maxn], n;
int lowbit(int x) { return x&-x; }

void update(int pos, int val)
{
    while(pos < maxn)
    {
        tree[pos] += val;
        pos += lowbit(pos);
    }
}

int query(int pos)
{
    int sum = 0;
    while(pos)
    {
        sum += tree[pos];
        pos -= lowbit(pos);
    }
    return sum;
}

int main(void)
{
    int t;
    cin >> t;
    while(t--)
    {
        scanf("%d", &n);
        for(int i = 1; i <= n; i++)
            scanf("%d", &a[i]);
        memset(tree, 0, sizeof(tree));
        for(int i = 3; i <= n; i++)
            if(a[i-2] && a[i-1] && !a[i])
                update(i-1, 1);
        if(a[n] && a[n-1]) update(n, 1);
        int q, l, r;
        scanf("%d", &q);
        while(q--)
        {
            scanf("%d%d", &l, &r);
            if(l == r) { puts("0"); continue; }
            int ans = query(r)-query(l-1);
            //因为下面两条是要在r-1、l+1的条件下使用的，所以l==r时不能用需要加特判
            if(a[r] && a[r-1] && a[r+1]) ans++;
            if(a[l-1] && a[l] && !a[l+1]) ans--;
            printf("%d\n", ans);
        }
    }
    return 0;
}