poj - 3067 - Japan（树状数组）

题意：日本东海岸有N个城市（从北到南命名为1, 2, ..., N），西海岸有M个城市（从北到南命名为1, 2, ..., M），东西之间有K条高速公路，问这K条高速公路有多少个交叉点（一个交叉点有且只有两条高速公路经过）（1000 <= N, M <= 1000）。

题目链接：http://poj.org/problem?id=3067

——>>好题。。。#^_^

先将东岸城市从北到南排序，再将西岸城市从北到南排序；接着从北到南遍历西岸城市，设a[i]表示西岸城市i南部的所有城市目前出现的次数，那么此时西岸城市i产生的交点数为a[i]，设sum[i]表示西岸前i个城市此时出现次数的和，则a[i] = sum(M) - sum(i)，于是可用树状数组了。。。#^_^

坑点：64位整数！！！

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

using namespace std;

const int maxn = 1000 + 10;

int T, N, M, K;
long long c[maxn], cro[maxn];
pair<int, int> edges[maxn*maxn];

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

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

void add(int x) {
    while(x <= M) {
        c[x]++;
        x += lowerbit(x);
    }
}

int main()
{
    int e, w;
    scanf("%d", &T);
    for(int kase = 1; kase <= T; kase++) {
        memset(c, 0, sizeof(c));
        memset(cro, 0, sizeof(cro));
        scanf("%d%d%d", &N, &M, &K);
        for(int i = 0; i < K; i++) {
            scanf("%d%d", &e, &w);
            edges[i] = make_pair(e, w);
        }
        sort(edges, edges+K);
        for(int k = 0; k < K; k++) {
            int x = edges[k].second;
            cro[x] += sum(M) - sum(x);
            add(x);
        }
        long long ret = 0;
        for(int i = 0; i < M; i++) ret += cro[i];
        printf("Test case %d: %I64d\n", kase, ret);
    }
    return 0;
}