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 @ 2023-11-05 11:57  蒟蒻orz  阅读(5)  评论(0编辑  收藏  举报  来源