谜一样的牛（acwing）（树状数组+二分）

ddgo

于 2021-06-30 22:07:30 发布

阅读量132

点赞数

分类专栏： 0开始的数据结构

本文链接：https://blog.youkuaiyun.com/qq_45604735/article/details/118370619

版权

0开始的数据结构专栏收录该内容

15 篇文章

订阅专栏

本文介绍了一种利用树状数组与二分查找解决特定类型问题的方法。该问题要求计算一系列数值，并通过二分查找及树状数组实现高效处理。文章详细解释了算法流程并提供了完整的代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目地址
 配套地址

在这里插入图片描述

/*
    从后往前，每个位置(i)的值应该为，
    将i后面所有确定好的数插入到数列中，第(A[i]+1)个空位对应的下标。
    可以用二分加树状数组。
    树状数组处理上述语句的前半段，二分处理后半段，去快速查找那个位置。
*/

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;

int c[N],n,a[N],ans[N];
bool vis[N];
#define lowbit(x) (x & -x)

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

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

int main(){
    cin>>n;
    for(int i=2;i<=n;i++) cin>>a[i];
    for(int i=n;i>=2;i--){
        int l = 1,r = n,k = a[i] + 1;
        while(l < r){
            int mid = (l + r) >> 1;
            if((mid - ask(mid)) >= k) r = mid;
            else l = mid + 1;
        }
        change(r,1);
        ans[i] = r;
        vis[r] = true;
    }
    ans[1] = 1;
    while(vis[ans[1]]) ans[1] ++;
    for(int i=1;i<=n;i++) cout<<ans[i]<<endl;
    return 0;
}