求栈弹出弹入后的最终次序的问题

最新推荐文章于 2024-12-11 01:36:35 发布

butane123

最新推荐文章于 2024-12-11 01:36:35 发布

阅读量101

点赞数

文章标签：算法

本文链接：https://blog.youkuaiyun.com/CAOSHUCAOSHU/article/details/119210560

版权

本文解析了通过连续栈操作将一个栈元素转移至另一个栈的过程，利用头尾数组和深度优先搜索算法追踪数字间的连接，展示了如何查询最终各栈数字链路。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

原题链接
每次操作输入a,b，表示将a栈中数全部弹出弹入到b栈中。求最终各栈的数字情况。
分析：
用head数组记录各栈的栈顶元素，tail数组记录栈底元素。对于每次操作，将head[a]与head[b]连接一条双向边，再更新head[b]为tail[a]即可。最终可以通过dfs(tail[i])，以此查询每个数字与其相连的数字情况，从而得到答案。

int head[maxv],tail[maxv];
vector<int> g[maxv],ans;
bool vis[maxv];
void dfs(int x){
    if(x==-1) return;
    ans.push_back(x);
    for(int i=0;i<g[x].size();i++) if(!vis[g[x][i]]) vis[g[x][i]]=1,dfs(g[x][i]),vis[g[x][i]]=0;
}
int main(){
    int n,m,a,b;
    while(scanf("%d %d",&n,&m)!=EOF){
        for(int i=1;i<=n;i++) head[i]=tail[i]=i,g[i].clear();
        for(int i=1;i<=m;i++){
            scanf("%d %d",&a,&b);
            if(head[a]==-1) continue;
            if(head[b]==-1) tail[b]=head[a],head[b]=tail[a];
            else g[head[b]].push_back(head[a]),g[head[a]].push_back(head[b]),head[b]=tail[a];
            head[a]=tail[a]=-1;
        }
        for(int i=1;i<=n;i++){
            ans.clear();
            vis[tail[i]]=1,dfs(tail[i]),vis[tail[i]]=0;
            printf("%d",ans.size()),reverse(ans.begin(),ans.end());
            for(int j=0;j<ans.size();j++) printf(" %d",ans[j]);
            printf("\n");
        }
    }
    return 0;
}