CF1889D Game of Stacks
当每个栈只有一个元素时,我们连接 , 为第 个栈的栈顶。
现在是一棵内向基环树了。
此时询问的答案是每个点对应的树的处在环上的根。
回到原问题,其实我们不断消环,即找到环,将环上的栈顶弹出,并新建连边 ,并维护每个点的根。
参考代码理解。
int n, tp, st[_], fg[_], vis[_];
stack<int> s[_];
int dfs(int u) {
if(fg[u]) return fg[u];
if(s[u].empty()) return u;
if(vis[u]) {
while(1) {
int nw = st[tp];
--tp;
s[nw].pop();
vis[nw] = 0;
if(nw == u) break;
}
return dfs(u);
}
st[vis[u] = ++tp] = u;
return dfs(s[u].top());
}
bool Med;
signed main() {
// Mry;
n = ri();
F(i, 1, n) {
int k = ri();
while(k--) {
int c = ri();
s[i].push(c);
}
}
F(i, 1, n) {
tp = 0;
int ok = dfs(i);
F(j, 1, tp) fg[st[j]] = ok;
printf("%d%c", ok, " \n"[i == n]);
}
// Try;
return 0;
}
本文来自博客园,作者:蒟蒻orz,转载请注明原文链接:https://www.cnblogs.com/orzz/p/18121918