OI竞赛中手工栈的书写

来源:互联网 发布:linux 打印环境变量 编辑:程序博客网 时间:2024/05/13 08:58

引言

在某些算法中,尤其是树、图、数据结构相关的算法,会牵扯到大量的递归。在绝大部分的OI竞赛中(NOIP、NOI等),递归所占用的栈空间限制为内存限制,换句话说,就是一般只要你不MLE就不会爆栈。
但是SDOI历年使用Windows+Cena评测,这个古老的评测系统递归的栈空间有限,经常递归到3w+就会爆栈,也就是说,如果你想遍历一条10w级别的树链是不可能的。
所以SDOI的选手们就开始写手工栈这个东西。。什么是手工栈呢?其实就是根据递归的原理用一个栈代替递归的栈空间,然后进行的操作什么的是不变的。本文将给出一些常用算法手工栈的书写。

概述

手工栈的主要部分和大体写法如下:
1、stack:手工栈。“手工栈”实际上就是这个数组,作为一个栈存储的是当前遍历到的所有点,模拟递归的过程。某一个点在栈内说明它并没有利用完成,某一个点已经出栈说明其所有的信息都已经利用过了。
2、cur:当前弧。数据结构中存储边的最常用方法是nxt数组(链表),而cur表示的就是这个点当前已经遍历到了它的哪一条边。会isap的同学可以发现这个和isap的当前弧是差不多的。
3、use:标记数组。有时需要用有时不需要用。有的递归过程在某一个点第一次遍历到的时候需要记录一些信息(比如dfs序等),这个时候就需要用到标记数组判断当前点是否是第一次遍历到。

算法流程:
1、将第一个点入栈
2、只要栈非空就一直访问栈顶
3、当栈顶的点所有信息都已经被利用就弹栈
注意:需要统计信息的时刻无非:某个点入栈、某个点第一次被访问、某个点出栈,在相对应的时刻统计信息就行了。

代码可以参考下文

树链剖分+dfs序

例题:BZOJ4034
就是一道链剖+dfs序裸题

在树链剖分和dfs序相关算法中,预处理会遍历整棵树,记录一些信息:父亲、重儿子、链顶、dfs序等。。

递归写法

void dfs_1(int x,int fa){    father[x]=fa;size[x]=1;h[x]=h[fa]+1;    for (int i=point[x];i;i=nxt[i])        if (v[i]!=fa)        {            dfs_1(v[i],x);            size[x]+=size[v[i]];            if (size[v[i]]>size[son[x]]) son[x]=v[i];        }}void dfs_2(int x,int fa){    if (son[fa]==x) top[x]=top[fa];    else top[x]=x;    in[x]=++dfs_clock;num[dfs_clock]=a[x];    if (son[x]) dfs_2(son[x],x);    for (int i=point[x];i;i=nxt[i])        if (v[i]!=fa&&v[i]!=son[x])            dfs_2(v[i],x);    out[x]=dfs_clock;}

手工栈写法

void dfs_1(){    int tmp=0;stack[++tmp]=1;    size[1]=1;h[1]=1;    for (int i=1;i<=n;++i) cur[i]=point[i];    while (tmp)    {        int x=stack[tmp];        while (cur[x]&&v[cur[x]]==father[x]) cur[x]=nxt[cur[x]];        if (!cur[x])        {            --tmp;            if (father[x])            {                size[father[x]]+=size[x];                if (size[x]>size[son[father[x]]]) son[father[x]]=x;            }            continue;        }        int vt=v[cur[x]];stack[++tmp]=vt;        father[vt]=x;size[vt]=1;h[vt]=h[x]+1;        cur[x]=nxt[cur[x]];    }}void dfs_2(){    int tmp=0;stack[++tmp]=1;    top[1]=1;in[1]=++dfs_clock;num[dfs_clock]=a[1];    for (int i=1;i<=n;++i) cur[i]=point[i];    while (tmp)    {        int x=stack[tmp];        if (!use[x])        {            use[x]=1;            if (son[x])            {                int vt=son[x];                top[vt]=top[x];in[vt]=++dfs_clock;num[dfs_clock]=a[vt];                stack[++tmp]=vt;            }            continue;        }        while (cur[x]&&(v[cur[x]]==father[x]||v[cur[x]]==son[x])) cur[x]=nxt[cur[x]];        if (!cur[x])        {            --tmp;            out[x]=dfs_clock;            continue;        }        int vt=v[cur[x]];stack[++tmp]=vt;        top[vt]=vt;in[vt]=++dfs_clock;num[dfs_clock]=a[vt];        cur[x]=nxt[cur[x]];    }}

完整代码 戳这里

Tarjan

例题 BZOJ1051

如果要是出题人出一条链来卡你的话也是会爆栈的。。
主要是tarjan本来就需要用栈好烦啊

递归写法

void tarjan(int x){    dfn[x]=low[x]=++dfs_clock;vis[x]=1;st[++t]=x;    for (int i=point[x];i;i=nxt[i])        if (!dfn[v[i]])        {            tarjan(v[i]);            low[x]=min(low[x],low[v[i]]);        }        else if (vis[v[i]])            low[x]=min(low[x],dfn[v[i]]);    if (dfn[x]==low[x])    {        int now=0;++scc;        while (now!=x)        {            now=st[t--];            belong[now]=scc;            ++cnt[scc];            vis[now]=0;        }    }}

手工栈写法

void tarjan(int x){    int tmp=0;t=0;stack[++tmp]=x;    dfn[x]=low[x]=++dfs_clock;vis[x]=1;st[++t]=x;cur[x]=point[x];    while (tmp)    {        int x=stack[tmp];        if (!cur[x])        {            --tmp;            if (dfn[x]==low[x])            {                int now=0;++scc;                while (now!=x)                {                    now=st[t--];                    belong[now]=scc;                    ++cnt[scc];                    vis[now]=0;                }            }            if (father[x]) low[father[x]]=min(low[father[x]],low[x]);            continue;        }        int vt=v[cur[x]];        if (!dfn[vt])        {            stack[++tmp]=vt;st[++t]=vt;            dfn[vt]=low[vt]=++dfs_clock;            father[vt]=x;vis[vt]=1;            cur[vt]=point[vt];        }        else if (vis[vt]) low[x]=min(low[x],dfn[vt]);        cur[x]=nxt[cur[x]];    }}

完整代码 戳这里

二分图匹配

例题 BZOJ1854
这种需要返回参数的东西有点GG,自己YY了一个有点麻烦的写法。。。

递归写法

bool find(int x,int k){     for (int i=point[x];i;i=nxt[i])        if (vis[v[i]]!=k)        {            vis[v[i]]=k;            if (!belong[v[i]]||find(belong[v[i]],k))            {                belong[v[i]]=x;                return true;            }        }    return false;}

手工栈写法

void find(int x,int k){    int top=0;stack[++top]=x;    cur[x]=point[x];    while (top)    {        int x=stack[top];        if (flag[belong[v[cur[x]]]]==k)        {            belong[v[cur[x]]]=x;            flag[x]=k;            --top;            continue;        }        while (cur[x]&&vis[v[cur[x]]]==k) cur[x]=nxt[cur[x]];        if (!cur[x])        {            --top;            continue;        }        int vt=v[cur[x]];        vis[vt]=k;        if (!belong[vt])        {            belong[vt]=x;            flag[x]=k;            --top;            continue;        }        cur[belong[vt]]=point[belong[vt]];        stack[++top]=belong[vt];    }}

完整代码 戳这里

点分治

感觉点分写手工栈最蛋疼啊。。递归的次数太多要写3次。。

递归写法

手工栈写法

0 0
原创粉丝点击