POJ 2182 Lost Cows

最新推荐文章于 2022-09-03 21:45:05 发布

Infinite_Jerry

最新推荐文章于 2022-09-03 21:45:05 发布

阅读量151

点赞数 1

分类专栏： # 线段树/平衡树 # 树状数组

本文为博主原创文章，未经博主允许不得转载。一经发现，必将追究法律责任。

本文链接：https://blog.csdn.net/qq_42886072/article/details/102526534

版权

线段树/平衡树同时被 2 个专栏收录

10 篇文章 0 订阅

订阅专栏

树状数组

8 篇文章 0 订阅

订阅专栏

【题意】
有n头奶牛，已知它们的身高为 1~n 且各不相同，但不知道每头奶牛的具体身高。
现在这n头奶牛站成一列，已知第i头牛前面有Ai头牛比它低，求每头奶牛的身高。

一道裸题.很明显从后往前扫,然后找到未选的数中第a[i]+1大即可.

方法1:权值线段树

在权值线段树上二分,并顺便删除.
需要 $5 n$ 内存.时间复杂度 $\log n)$
代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
#define lc (x<<1)
#define rc (x<<1|1)
using namespace std;
const int N=1e5+10;
int s[N<<2];
void bt(int x,int l,int r) {
	s[x]=r-l+1;
	if(l==r)return;
	int mid=(l+r)>>1;
	bt(lc,l,mid);
	bt(rc,mid+1,r);
}
int ans;
void find(int x,int l,int r,int k) {
	s[x]--;
	if(l==r){ans=l;return ;}
	int mid=(l+r)>>1;
	if(k<=s[lc])find(lc,l,mid,k);
	else 		find(rc,mid+1,r,k-s[lc]);
}
int n,a[N];
int main() {
	scanf("%d",&n);bt(1,1,n);
	for(int i=2;i<=n;i++)scanf("%d",&a[i]);
	for(int i=n;i	;i--) {
		find(1,1,n,a[i]+1);
		a[i]=ans;
	}
	for(int i=1;i<=n;i++)printf("%d\n",a[i]);
	return 0;
}

方法2:树状数组+倍增

设b维护前缀可选位置的个数.(树状数组)
从后往前扫,对于每个位置 $i$ ,步骤如下:

初始化 $a n s = 0, s u m = 0$
从 $\lfloor{\log n}\rfloor\sim0$ 枚举 $j$ ,若 $ans+2^j\le n且sum+b[ans+2^j]\le a[i]$
$ans+=2^j,sum+=b[ans]$ .
$a [i] = a n s + 1$ (记录答案), $d e l (a [i])$

步骤2非常巧妙的一点是由于指数从大到小枚举 $ans+2^j的lowbit恰好为2^j,那么(ans,ans+2^j]的可放位置个数就是b[ans+2^j]$

这个算法只需要 $2 n$ 内存.时间复杂度 $\log n)$ (常数都比线段树小)

代码:

#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1e5+10;
int n,t,a[N],b[N];
void del(int x) {for(	;x<=n;x+=x&-x)b[x]--;}
int main() {
	scanf("%d",&n);
	t=log2(n);b[1]=1;
	for(int i=2;i<=n;i++)scanf("%d",&a[i]),b[i]=i&-i;
	for(int i=n;i	;i--) {
		int ans=0,sum=0;
		for(int j=t;j>=0;j--)
			if(ans+(1<<j)<=n&&sum+b[ans+(1<<j)]<=a[i])	
				ans+=1<<j,sum+=b[ans];
		del(a[i]=ans+1);
	}
	for(int i=1;i<=n;i++)printf("%d\n",a[i]);
	return 0;
}