IT博客-魔のkyo的工作室-随笔分类-DataStucture

Union Find Set

魔のkyo — Sun, 21 Oct 2007 07:51:00 GMT

const int MaxSize = 100000;

class UFSets {
private:
    int parent[MaxSize+1];
    int size;
public:
    UFSets (int s ){
        size = s;
        memset(parent,-1,sizeof(parent));
    }

    int Find (int x){
        if ( parent[x] <0 ) return x;
        else return Find (parent[x]);
    }

    void   Union (int v1, int v2){
        int s1=Find(v1),s2=Find(v2);
        if(s1==s2)return;
        int t = parent[s1] + parent[s2];
        if ( parent[s2] < parent[s1] ) {
            parent[s1] = s2;
            parent[s2] = t;
        }
        else {
            parent[s2] = s1;
            parent[s1] = t;
        }
    }

    int Count (int x){
        return -parent[Find(x)];
    }
};

魔のkyo 2007-10-21 15:51 发表评论

线段树Segment Tree

魔のkyo — Sat, 25 Aug 2007 18:31:00 GMT

//线段树,添加，删除，查询排名，查询第k小数都在O(logn)级，O(NlogN)时间内可求出一个排列的逆序数 example 1484，2425
//http://www.cnitblog.com/cockerel/archive/2006/09/13/16806.html
template <int N> //表示可用区间为[0,N)，其中N必须是2的幂数;
class PointTree{
    int a[2*N];
    int size;
public:
    void clear(){memset(this,0,sizeof(*this));}

    void add(int n){
        int i=N+n;
        ++size;
        for(++a[i];i>1;i/=2)
            if(~i&1)a[i/2]++ ;
    }

    int cntEQ(int n){return a[N+n];}

    int cntLs(int n){ //统计小于
        int i=N+n,c=0;  //若统计小于等于则c=a[i];
        for (; i>1 ; i/=2)
            if (i&1) c += a[i/2];
            return c;
    }

    int cntGt(int n){return  size-a[N+n]-cntLs(n);}

    void del(int n){
        if(!a[n+=N])return;
        --size;
        for(--a[n]; n>1; n/=2)
            if(~n&1)--a[n/2];
    }
    /*解决：求点集中第i小的数(由0数起)
      注意：如果i>=size 返回N-1 */
    int operator[](int n){
        int i=1;
        while(i<N){
            if(n<a[i]) i*=2;
            else n-=a[i], i=i*2+1;
        }
        return i-N;
    }
};

魔のkyo 2007-08-26 02:31 发表评论

树状数组Binary Indexed Tree

魔のkyo — Sat, 25 Aug 2007 18:29:00 GMT

//Binary Indexed Tree
//以O(logn)的时间完成修改项，查询前n项和

template <int N>//表示可用区间为[1,N]，其中N必须是2的幂
class BITree{
    int C[N+1];
    int lowbit(int i)
    {
        return i&(-i);
    }
public:
    BITree(){memset(C,0,sizeof(C));}
    void clear(){memset(C,0,sizeof(C));}

    void modify(int i,int d)
    {
        while(i<=N){
            C[i]+=d;
            i+=lowbit(i);
        }
    }

    //返回sum[i]=sum{a1,a2,,ai}
    int operator [] (int i)
    {
        int sum=0;
        while(i){
            sum+=C[i];
            i-=lowbit(i);
        }
        return sum;
    }
};

魔のkyo 2007-08-26 02:29 发表评论

用C++实现滚动数组

魔のkyo — Tue, 31 Jul 2007 16:45:00 GMT

很多DP都可以节省内存，一般通过滚动数组的方法，下面是一个M行的滚动数组，可以看到实现得非常优雅

const int M=2;
double _dp[M][100000+1];

struct{
    double* operator [](int i){
        return _dp[i%M];
    }
}dp;

魔のkyo 2007-08-01 00:45 发表评论