lost cows

2020-02-12 16:00:45来源:博客园 阅读 ()

新老客户大回馈,云服务器低至5折

lost cows

这道题用树状数组做比较好,虽然树状数组能做的线段树也可以做到,但是树状数组更简洁方便,易操作

原理便是第x个数的二进制数最后一个“1”,决定tree的结点的长度

比如:

sum[3]=tree[3]+tree[2];

sum[4]=tree[4];

sum[5]=tree[5]+tree[4];

分割是位运算里的操作,一个数和它负数的补码(取反+1)用x&-x求得二进制数中为“1”的最小位

代码如下:

#include <iostream>
#include <cstdio>
#define lowbit(x) ((x)&-(x))
using namespace std;
const int MAX=10000;
int n;
int tree[MAX],pre[MAX],ans[MAX];


void add(int x,int d){//给x加d
while(x<=n){
tree[x]+=d;
x+=lowbit(x);
}
}

 

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


int findpos(int x){
int l=1,r=n;
while(l<r){
int mid=(l+r)>>1;
if(sum(mid)<x){
l=mid+1;
}
else{
r=mid;
}
}
return l;
}


int main()
{
scanf("%d",&n);
pre[1]=0;
for(int i=2;i<=n;i++){
scanf("%d",&pre[i]);
}
for(int i=1;i<=n;i++){//初始化不需要add()
tree[i]=lowbit(i);
}
for(int i=n;i>0;i--){
int x=findpos(pre[i]+1);
add(x,-1);
ans[i]=x;
}
for(int i=1;i<=n;i++){
printf("%d\n",ans[i]);
}
return 0;
}


原文链接:https://www.cnblogs.com/sos3210/p/12298452.html
如有疑问请与原作者联系

标签:

版权申明:本站文章部分自网络,如有侵权,请联系:west999com@outlook.com
特别注意:本站所有转载文章言论不代表本站观点,本站所提供的摄影照片,插画,设计作品,如需使用,请与原作者联系,版权归原作者所有

上一篇:游戏终端服务器分享(1.2版)

下一篇:学过 C++ 的你,不得不知的这 10 条细节!