现在的位置: 首页 > 算法 > 正文

poj – 3067 – Japan(树状数组)

2019年08月29日 算法 ⁄ 共 1034字 ⁄ 字号 评论关闭

题意:日本东海岸有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;
}

抱歉!评论已关闭.