POJ 2352 树状数组(线段树)

题意:给出n个点的坐标(n<=15000),坐标范围为[1,32000],最后让你输出ans[i],表示有ans[i]个点左下方有i个点。

一开始直接写的二维树状数组,记录一个矩阵,如果(x,y)有点就是1,没有点就是0,那么我们可以轻易求出(1,1)到(x,y)共有多少点,不过这样就爆了10倍内存。(1K == 256 int型整数)

然而后来才注意到:题目输入是按照y坐标从小到大输入的。(英语水平很重要TAT)
y坐标按照从小到大输入,所以可以只用一维的树状数组,只需要记录横坐标上1到x有多少个点即可。需要离散化,一开始写的没有离散化的T掉了。
代码:

#include <cstdio>
#include <algorithm>
#include <cstring>
#define lowbit(x) x&-x
using namespace std;

int n, m, ans[15005], x[15005], y, f[32005];

struct star{
    int x, num;
}t[15005];
bool cmp(star a, star b){
    return a.x < b.x;
}

void add(int x){
    while(x <= n) f[x]++, x += lowbit(x);
}

int sum(int x){
    int ans = 0;
    while(x) ans += f[x], x -= lowbit(x);
    return ans;
}

int main()
{
    scanf("%d", &n);
    for(int i = 1; i <= n; i++){
        scanf("%d %d", &t[i].x, &y);
        t[i].num = i;
    }

    sort(t+1, t+n+1, cmp);
    int cnt = 1;
    x[t[1].num] = 1;
    for(int i = 2; i <= n; i++){
        if(t[i].x == t[i-1].x) x[t[i].num] = cnt;
        else x[t[i].num] = ++cnt;
    }

    for(int i = 1; i <= n; i++){ //☆☆
        add(x[i]);
        ans[sum(x[i])-1]++;
    }

    for(int i = 0; i < n; i++)
        printf("%d\n", ans[i]);
}

有细节需要注意的是,空间用一维树状数组的区别就是代码里标五角星的那个for循环,如果是二维的话应该是这么写的:

    for(int i = 1; i <= n; i++)
        add(t[i].x, t[i].y);
    for(int i = 1; i <= n; i++){
        int x = t[i].x, y = t[i].y, k;
        k = sum(x-1, y) + sum(x, y-1) - sum(x-1, y-1);
        ans[k]++;
    }

这个需要实际理解一下左下方的含义,(x,y)的左下方的意思是(1,1)到(x,y)的矩形内的范围(除了自己),所以在一维的计算中是ans[sum(x[i])-1]++, 二维就是sum(x-1, y) + sum(x, y-1) - sum(x-1, y-1) (当然二维也可以是sum(x,y)-1),二维的两种写法都很好理解,一维的就只有那一种写法(sum(x[i]-1)是不对的,因为有可能会有x坐标相同的,y坐标比当前点小的点,而它们在一维的树状数组中都被记录成同一个x坐标),需要好好体会一下。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
POJ 2182是一道使用树状数组解决的题目,题目要求对给定的n个数进行排序,并且输出每个数在排序后的相对位置。树状数组是一种用来高效处理前缀和问题的数据结构。 根据引用中的描述,我们可以通过遍历数组a,对于每个元素a[i],可以使用二分查找找到a到a[i-1]中小于a[i]的数的个数。这个个数就是它在排序后的相对位置。 代码中的query函数用来求前缀和,add函数用来更新树状数组。在主函数中,我们从后往前遍历数组a,通过二分查找找到每个元素在排序后的相对位置,并将结果存入ans数组中。 最后,我们按顺序输出ans数组的元素即可得到排序后的相对位置。 参考代码如下: ```C++ #include <iostream> #include <cstdio> using namespace std; int n, a += y; } } int main() { scanf("%d", &n); f = 1; for (int i = 2; i <= n; i++) { scanf("%d", &a[i]); f[i = i & -i; } for (int i = n; i >= 1; i--) { int l = 1, r = n; while (l <= r) { int mid = (l + r) / 2; int k = query(mid - 1); if (a[i > k) { l = mid + 1; } else if (a[i < k) { r = mid - 1; } else { while (b[mid]) { mid++; } ans[i = mid; b[mid = true; add(mid, -1); break; } } } for (int i = 1; i <= n; i++) { printf("%d\n", ans[i]); } return 0; } ``` 这段代码使用了树状数组来完成题目要求的排序功能,其中query函数用来求前缀和,add函数用来更新树状数组。在主函数中,我们从后往前遍历数组a,通过二分查找找到每个元素在排序后的相对位置,并将结果存入ans数组中。最后,我们按顺序输出ans数组的元素即可得到排序后的相对位置。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *3* [poj2182Lost Cows——树状数组快速查找](https://blog.csdn.net/aodan5477/article/details/102045839)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] - *2* [poj_2182 线段树/树状数组](https://blog.csdn.net/weixin_34138139/article/details/86389799)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值