51nod1533 一堆的堆

1533 一堆的堆 

题目来源：  CodeForces

基准时间限制：1.5 秒 空间限制：131072 KB 分值: 80  难度：5级算法题

 收藏

 关注

现在有一个长度为n的数组 a1, a2, ..., an 。然后对于k从1到 n-1分别对该数组建k叉堆。现在要统计对于每一个k叉堆，里面有多少结点是不满足最小堆的性质的。即值比父亲的要小的结点有多少个。

k叉堆的定义是这样的：数组的下标从1到n编号，对于某一个编号为v的结点，他的k个儿子编号是 k(v − 1) + 2, ..., kv + 1 （如果其中某些编号超出n，那些编号就不要）。在k叉堆中，除了根以外，每一个结点都有一个父亲。p(v)表示结点v的父亲。那么一个非根结点不满足最小堆性质当且仅当 av < ap(v) 

样例解释：

下面四个图分别对应k=1,2,3,4的情况。

非法结点用红色标记。

Input

单组测试数据。
第一行有一个整数 n (2≤n≤2*10^5)。
第二行有n个用空格分开的整数a1, ..., an (-10^9≤ai≤10^9)。

Output

输出n-1个数字，第k个数字代表k叉堆中不满足最小堆性质的结点数目 （k=1, 2, ..., n-1）。

Input示例

样例输入1
5
1 5 4 3 2

Output示例

样例输出1
3 2 1 0

题解：

在kk-ary heap中，i的儿子是k*(i-1)+2到k*i+1这一段区间。

那么ii有儿子的kk的取值大约有ni−1ni−1中，对于每一个ii都枚举kk，复杂度是O(nlogn)O(nlogn)的。

我们对原数组排序，从小到大加入树状数组，每次枚举kk，统计儿子中比他小的个数即可。

注意，对于数值相同的情况，把位置靠前的放在前面即可。

代码：

#include<bits/stdc++.h> 
using namespace std;
struct aaa{
	int aa,bb;
}a[1000000];
int n,l,r,b[1000000],ans[1000000];
bool cmp(aaa a,aaa b){
	return a.aa<b.aa||a.aa==b.aa&&a.bb<b.bb;
}
int query(int x){
	int ans=0;
	for(int i=x;i;i-=i&(-i))ans+=b[i];
	return ans;
}
int insert(int x){
	for(int i=x;i<=n;i+=i&(-i))b[i]++;
}
int main(){
	int i,p,j,l,r;
	scanf("%d",&n);
	for(i=1;i<=n;i++){
		scanf("%d",&a[i].aa);
		a[i].bb=i;
	}
	sort(a+1,a+n+1,cmp);
	for(i=1;i<=n;i++){
		p=a[i].bb;
		for(j=1;j*(p-1)+2<=n&&j<n;j++){
			l=(p-1)*j+2;r=min(n,j*p+1);
			ans[j]+=query(r)-query(l-1);
		}
		insert(p);
	}
	for(i=1;i<n;i++)printf("%d ",ans[i]);
}