ACM模板 数据结构

一、并查集

const int maxn = 30000+5;
int n;
int par[maxn], rk[maxn];//par记录祖先,rk记录其在所在树中的深度
void init()
{
    for(int i = 0; i < n; i++)
        par[i] = i, rk[i] = 0;
}

int finda(int x)
{
    if(par[x] == x)  return x;
    return par[x] = finda(par[x]);
}

void unite(int x, int y)
{
    x = finda(x);
    y = finda(y);
    if(x == y) return;

    if(rk[x] < rk[y]) par[x] = y;
    else
    {
        par[y] = x;
        if(rk[x] == rk[y]) rk[x]++;
    }
}

二、树状数组

注意树状数组中最小的数为1，不能是0，当输入数据最小值为0时，可全体加一
树状数组初始化为0

1.一维单点更新、区间查询

一维单点更新、区间求和
maxb为所用最大数字，要初始化

int lowbit(int x) {return x&-x;}

void add(int x,int val)
{for(int i=x;i<=maxb;i+=lowbit(i)) bit[i]+=val;}

int getsum(int x)
{
    int res=0;
    for(int i=x;i>0;i-=lowbit(i)) res+=bit[i];
    return res;
}

int query(int l,int r)
{return getsum(r)-getsum(l-1);}

例题：
poj 2299 Ultra-QuickSort
poj 3321 Apple Tree
poj 3067 Japan
poj 3416 Crossing

2.一维区间更新、单点查询

此时记录每个点与前面一点的增值，则查询某 点值即为从头到这个点的增值的累和，即由getsum()得到。
更新区间如下：

void update(int l,int r,int val)
{
    add(l,val);
    add(r+1,-val);
}

3.二维单点更新、区间查询

二维需要用容斥原理得到待求的二维区间

inline int lowbit(int x){return x&-x;}
void add(int x,int y,int val)
{
    for(int i=x;i<=maxb;i+=lowbit(i))
        for(int j=y;j<=maxb;j+=lowbit(j))
            bit[i][j]+=val;
}
int getsum(int x,int y)
{
    int res=0;
    for(int i=x;i>0;i-=lowbit(i))
        for(int j=y;j>0;j-=lowbit(j))
            res+=bit[i][j];
    return res;
}
int calc(int x1,int y1,int x2,int y2)
{
    return getsum(x2,y2)-getsum(x1-1,y2)-getsum(x2,y1-1)+getsum(x1-1,y1-1);
}

例题：poj1195 Mobile phones

4.二维区间更新、单点查询

将一维区间更新、单点查询改为二维，同样记录的是增值。如图，更新黄色区域，用容斥原理，双斜线处增加，单斜线处减少。
    1处增加后，2、3两处再减回去，到了4被减了两遍，再加回来。

int lowbit(int x){return x&-x;}
void add(int x,int y,int val)
{
    for(int i=x;i<=maxb;i+=lowbit(i))
        for(int j=y;j<=maxb;j+=lowbit(j))
            bit[i][j]+=val;
}
void update(int x1,int y1,int x2,int y2,int val)
{
    add(x1,y1,val);
    add(x1,y2+1,-val);
    add(x2+1,y1,-val);
    add(x2+1,y2+1,val);
}
int getsum(int x,int y)
{
    int res=0;
    for(int i=x;i>0;i-=lowbit(i))
        for(int j=y;j>0;j-=lowbit(j)) 
            res+=bit[i][j];
    return res;

例题：poj 2155 Matrix

5.[技巧]双关键字排序预处理

例题：
poj 2352 Stars
poj 2481 Cows
poj 3067 Japan
poj 3416 Crossing

三、线段树

1.单点更新、区间查询

单点更新、区间求和

#define lson l,mid,cur<<1
#define rson mid+1,r,cur<<1|1
const int maxn=50000+5;
int st[maxn<<2];//开n的4倍大小
int n;

inline void push_up(int cur)
{st[cur]=st[cur<<1]+st[cur<<1|1];}

void build(int l,int r,int cur)
//按节点从左至右的顺序初始化
//自顶向下，先build(1,n,1);
{
    if(l==r)
    {
        scanf("%d",&st[cur]);
        return;
    }
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
    push_up(cur);
}

void update(int p,int val,int l,int r,int cur)
//更新p点，使其增加val，当前在节点cur，对应区间[l,r]
//自顶向下更新，先update(p,val,1,n,1);
{
    if(l==r)
    {
        st[cur]+=val;
        return;
    }
    int mid=(l+r)>>1;
    if(p<=mid) update(p,val,lson);
    else update(p,val,rson);
    push_up(cur);
}

int query(int s,int t,int l,int r,int cur)
//要求sum(s,t),当前在代表[l,r]的点cur上
{
    if(s<=l && r<=t) return st[cur];
    if(t<l || s>r) return 0;
    int mid=(l+r)>>1;
    return query(s,t,lson)+query(s,t,rson);
}

例题：hdu 1166敌兵布阵

单点更新、区间求最值

下为求最大值,在上面类型1的基础上需要改:

• push_up()中做和改为用max
• update()中st[cur]+=val中的+=改为=
• query()中return的也用max
  如下所示：

inline void push_up(int cur)
{st[cur]=max(st[cur<<1],st[cur<<1|1]);}

void update(int p,int val,int l,int r,int cur)
{
    if(l==r)
    {
        st[cur]=val;
        return;
    }
    int mid=(l+r)>>1;
    if(p<=mid) update(p,val,lson);
    else update(p,val,rson);
    push_up(cur);
}

int query(int s,int t,int l,int r,int cur)
{
    if(s<=l && t>=r) return st[cur];
    if(t<l || s>r) return 0;
    int mid=(l+r)>>1;
    return max(query(s,t,lson),query(s,t,rson));
}

例题：hdu 1754 I Hate It

2.区间更新、区间查询

区间更新、区间求和