hdu -2836 - Traversal（树状数组）

题意：一个由N个正整数组成的序列，从中选出至少2个整数，使得相邻整数的差距不超过H，问总共有多少种选法（1 < N < 100001, 0 < H < 100000001, 序列中的正整数 <= 10^8）。

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

——>>设以第i个正整数h[i]结尾的选法为d，则d = 以值为[h[i] - H, h[i] + H]结尾的选法数的和，一个区间的连续和，可用树状数组来维护啦。即

d = ((sum(R) - sum(L-1) + 1) % mod + mod) % mod;

加1，是因为h[i]自己，不再选其他的整数，也符合[h[i] - H, h[i] + H]，题目要求至少2个数，最后的总和减去所有的单个自己就是答案。

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int maxn = 100000 + 10;
const int mod = 9901;
int h[maxn], N, H, buf[maxn], len, C[maxn];

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

int sum(int x){
    int ret = 0;
    while(x){
        ret += C[x];
        x -= lowerbit(x);
    }
    return ret;
}

void add(int x, int v){
    while(x <= N){
        C[x] += v;
        x += lowerbit(x);
    }
}

void read(){
    for(int i = 1; i <= N; i++) scanf("%d", &h[i]);
}

void init(){
    memcpy(buf, h, sizeof(h));
    sort(buf + 1, buf + 1 + N);
    len = unique(buf + 1, buf + 1 + N) - buf - 1;
    memset(C, 0, sizeof(C));
}

void solve(){
    int i, id, L, R, ret = 0, d;
    for(i = 1; i <= N; i++){
        id = lower_bound(buf + 1, buf + 1 + len, h[i]) - buf;
        L = lower_bound(buf + 1, buf + 1 + len, h[i] - H) - buf;
        R = upper_bound(buf + 1, buf + 1 + len, h[i] + H) - buf - 1;
        d = ((sum(R) - sum(L-1) + 1) % mod + mod) % mod;        //加1是h[i]自已
        add(id, d);
        ret += d;
    }
    ret = ((ret - N) % mod + mod) % mod;      //减去每个自己
    printf("%d\n", ret);
}

int main()
{
    while(scanf("%d%d", &N, &H) == 2){
        read();
        init();
        solve();
    }
    return 0;
}