POJ 2481Cows 树状数组

来源:互联网 发布:三岛由纪夫 禁色 知乎 编辑:程序博客网 时间:2024/05/29 17:18

题目:http://poj.org/problem?id=2481

题意:给定n个区间,问每个区间被多少个区间包含(自己不能包含自己)

思路:先按照右端点从大到小排序,然后左端点从小到大排序。然后查询每个左端点前面有几个数,就是可以包含当前区间的区间个数。注意特判两个区间相等时的情况

#include <iostream>#include <cstring>#include <cstdio>#include <algorithm>#include <vector>#include <cmath>using namespace std;typedef long long ll;const int N = 100010;struct node{    int x, y, id;}arr[N];int bit[N];int n, res[N];bool cmp(node a, node b){    if(a.y != b.y) return a.y > b.y;    else return a.x < b.x;}void add(int i, int x){    while(i <= 100001)        bit[i] += x, i += i & -i;}int sum(int i){    int s = 0;    while(i > 0)        s += bit[i], i -= i & -i;    return s;}int main(){    while(scanf("%d", &n), n)    {        memset(bit, 0, sizeof bit);        for(int i = 1; i <= n; i++)        {            scanf("%d%d", &arr[i].x, &arr[i].y);            arr[i].id = i;            arr[i].x++, arr[i].y++;        }        sort(arr + 1, arr + 1 + n, cmp);        for(int i = 1; i <= n; i++)        {            if(arr[i].x == arr[i-1].x && arr[i].y == arr[i-1].y)                res[arr[i].id] = res[arr[i-1].id];            else res[arr[i].id] = sum(arr[i].x);            add(arr[i].x, 1);        }        for(int i = 1; i <= n; i++)            printf("%d%c", res[i], i == n ? '\n' : ' ');    }    return 0;}


0 0
原创粉丝点击