hdu 5178想法题

比赛时，想法是：固定位置b,然后在b前面找大小处于[x[b] - k, x[b] + k]的数字个数，最后在加起来就是结果，但是要动态的维护b之前的数自动从小到大排序，每次都会新添加一个数，可以用二分插入，但是往后移动的数太多，直接就tle了。

后来的想法是：把原题等价转化一下，先把数据排序，然后对于当前位置i，找到左边的第一个>= x[i] - k的位置pos，则结果就是所有i - pos 的和。

原因是：排序后，x[b] - x[a] <= k， b > a。

讨论：b未排序的位置 > a未排序的位置，很显然位置落在b前面，求出的是[x[b] - k, x[b] + k]的做左半部分，即小于的情况，符合比赛时的想法。

b未排序的位置< a为排序的位置，很显然a落在b之后，看上去不符合原来的想法，其实这可以看做是0 >= x[a] - x[b] >= -k，这是符合比赛时的想法的，相当于求出[x[b] - k, x[b] + k]的右半部分，这也是合法的。

所以，左半部分加上去了，同时把后面的几个的右半部分也先算进来了，这是等价的。

lower_bound()返回的是第一个大于等于data[i] - k的位置，例如要在1，2，2，2，3，5中找2，则位置是1，找4，则位置是5。

#include<iostream>#include<algorithm>#include<cstdio>using namespace std;const int MAX = 100010;int data[MAX];int main(){    int t, n, k, i;    //freopen("in.txt", "r", stdin);    long long res;    scanf("%d", &t);    while(t--)    {        scanf("%d%d", &n, &k);        for(i = 0; i < n; i++)        {            scanf("%d", &data[i]);        }        sort(data, data + n);        res = 0;        for(i = 0; i < n; i++)        {            res = res + i - (lower_bound(data, data + n, data[i] - k) - data);        }        printf("%I64d\n", res);    }    return 0;}