AcWing 244. 谜一样的牛（树状数组+二分 | 树状数组的应用）

最新推荐文章于 2024-08-27 22:07:38 发布

QingQingDE23

最新推荐文章于 2024-08-27 22:07:38 发布

阅读量119

点赞数

分类专栏： # 树状数组数据结构文章标签： c++ 算法开发语言

本文链接：https://blog.csdn.net/qiaodxs/article/details/126795394

版权

数据结构同时被 2 个专栏收录

27 篇文章 1 订阅

订阅专栏

树状数组

8 篇文章 0 订阅

订阅专栏

本文详细介绍了如何利用树状数组和二分查找技术解决AcWing在线判题平台上的第244题。代码中展示了如何将普通问题转化为树状数组问题，并提供了完整的C++实现，包括lowbit函数、add函数和sum函数，以及主要的解题逻辑。通过对给定高度的牛进行处理，计算出每个位置的最小边界，最终输出答案。

摘要由CSDN通过智能技术生成

AcWing 244. 谜一样的牛
树状数组+二分，注意题中所引导出的怎么把普通问题转化为树状数组的问题

#include<bits/stdc++.h>

using namespace std;

#define int long long

const int N = 1e5 + 10;

int n, m;
int tr[N];
int h[N];
int ans[N];

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

void add(int x, int c){
	for(int i = x; i <= n; i += lowbit(i)){
		tr[i] += c;
	}
}

int sum(int x){
	int res = 0;
	for(int i = x; i; i -= lowbit(i)){
		res += tr[i];
	}
	return res;
}

signed main()
{
	cin>>n;
	for(int i = 2; i <= n; i ++ ) cin>>h[i];
	for(int i = 1; i <= n; i ++ ) add(i, 1);
	
	for(int i = n; i; i -- ){
		int k = h[i] + 1;
		int l = 1, r = n;
		while(l < r){
			int mid = (l + r) >> 1;
			if(sum(mid) >= k) r = mid;
			else l = mid + 1;
		}
		ans[i] = r;
		add(r, -1);
	}
	
	for(int i = 1; i <= n; i ++ ) cout<<ans[i]<<endl;
	
	return 0;
}