技术干货程序猿阵线联盟-汇总各类技术干货程序员

线段树 BIT 求冒泡排序的交换次数

2018-03-09  本文已影响49人  小太阳花儿

线段树特别适合与区间相关的运算,比如MRQ(minimum range query)求一段区间内的最小值。
BIT可以看作是压缩了的线段树,由于(某类)线段树的右节点可以由父结点和左兄弟求出来,所以右结点就不用存了。
求冒泡排序的交换次数,直观的想可以直接在冒泡排序的过程中计算交换次数,时间复杂度是O(n^2)。交换次数其实是(位置在j的前面,数值还比aj大的数)j从0到n求和。算法的特点是反复对某一区间内的所有元素求和,可以用BIT来优化。
做法是,i从0到n遍历,在循环的某一趟中,dat[ai]表示数字ai的之前出现过的次数,sum(ai)表示在这一趟之前,小于等于ai的数出现的次数。

#include <iostream>
using namespace std;

const int MAX_N = 10000;
int a[MAX_N],n;
int dat[1000000]; // 假定max(a[i])为1000000

//以下是对BIT的实现
int sum(int i){
    int ans = 0;
    while(i>0)
    {
        ans+=dat[i];
        i -= i&-i;
    }
    return ans;
}

void add(int i,int x){
    while(i<=n){
        dat[i] += x;
        i += i&-i;
    }
}

//计算冒泡排序的交换次数
int calu()
{
    cout<<"this is calu"<<endl;
    int ans = 0;
    for(int i=0;i<n;i++)
    {
        ans += i - sum(a[i]);
        add(a[i],1);
    }
    return ans;

}

int main()
{
    cin<<n;
    for(int i=0;i<n;i++)
    {
        cin>>a[i];
    }
    cout<<"ans is "<<calu()<<endl;
    return 0;

}
上一篇下一篇

猜你喜欢

热点阅读