CF1889D Game of Stacks

当每个栈只有一个元素时,我们连接 ipii\to p_ipip_i 为第 ii 个栈的栈顶。

现在是一棵内向基环树了。

此时询问的答案是每个点对应的树的处在环上的根。

回到原问题,其实我们不断消环,即找到环,将环上的栈顶弹出,并新建连边 ipii\to p_i,并维护每个点的根。

参考代码理解。

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;
}
posted @   蒟蒻orz  阅读(7)  评论(0编辑  收藏  举报  
相关博文:
阅读排行:
· .NET 10 首个预览版发布,跨平台开发与性能全面提升
· 全程使用 AI 从 0 到 1 写了个小工具
· 快收藏!一个技巧从此不再搞混缓存穿透和缓存击穿
· AI 插件第二弹,更强更好用
· Blazor Hybrid适配到HarmonyOS系统
历史上的今天:
2021-11-05 CF706D Vasiliy's Multiset
点击右上角即可分享
微信分享提示