poj3067 树状数组求逆序数

题意:日本岛东海岸与西海岸分别有N和M个城市,现在修高速公路连接东西海岸的城市,求交点个数。

做法:记每条告诉公路为(x,y), 即东岸的第x个城市与西岸的第y个城市修一条路。当两条路有交点时,

          满足(x1-x2)*(y1-y2)  < 0。所以,将每条路按x从小到达排序,若x相同,按y从小到大排序。                                                    然后按排序后的公路用树状数组在线更新,求y的逆序数之  和    即为交点个数。

         上面说的可能有点难理解,详细说明如下。

         记第i条边的端点分别为xi,yi。

         由于x是从小到大排序的,假设当前我们在处理第k条边,那么第1~k-1条边的x必然是小于(等于时候暂且不讨论)第k条边的    x 的,那么前k-1条边中,与第k条边相交的边的y值必然大于yk的,所以此时我们只需要求出在前k-1条边中有多少条边的y值在区间[yk, M]即可,也就是求yk的逆序数,M为西岸城市个数,即y的最大值。 所以就将问题转化成区间求和的问题,树状数组解决。

           当两条边的x相同时,我们记这两条边的y值分别为ya,yb(ya<yb),我们先处理(x,ya),再处理(x,yb),原因很明显,因为当x相同时,这两条边是认为没有交点的,若先处理(x,yb),那么下次处理(x,ya)时,(x,ya)就会给(x,yb)增加一个逆序,也就是将这两条边做相交处理了。

代码如下:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

#define M 1005

struct Node { int e, w; } road[M*M/2];
int c[M];

bool cmp(const Node& a, const Node& b) {
    if (a.e == b.e)
        return a.w < b.w;
    return a.e <= b.e;
}

inline __int64 lowbit(__int64 x) {
    return x & -x;
}

inline void update(__int64 x, __int64 delta, __int64 Max) {
    while (x <= Max) {
        c[x] += delta;
        x += lowbit(x);
    }
}

inline __int64 getSum(__int64 x) {
    __int64 sum = 0;
    while (x) {
        sum += c[x];
        x -= lowbit(x);
    }
    return sum;
}

int main()
{
    int i, j, T, n, m, q, total;
    __int64 ans;

    scanf ("%d", &T);
    for (i = 1; i <= T; i++) {
        memset(c, 0, sizeof (c));
        scanf ("%d%d%d", &n, &m, &q);
        for (j = 1; j <= q; j++)
             scanf ("%d%d", &road[j].e, &road[j].w);
        sort(road+1, road+1+q, cmp);
        ans = total = 0;
        for (j = 1; j <= q; j++) {
            ans += total - getSum(road[j].w);
            update(road[j].w, 1, m);
            total++;
        }
        printf("Test case %d: %I64d\n", i, ans);
    }
    return 0;
}


你可能感兴趣的:(c,struct)