记录编号 601502 评测结果 AAAAAAAAAA
题目名称 4164.[CF1889D] Game of Stack 最终得分 100
用户昵称 Gravatar健康铀 是否通过 通过
代码语言 C++ 运行时间 2.735 s
提交时间 2025-06-25 15:17:48 内存使用 78.70 MiB
显示代码纯文本
#include <bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int n,k[N],ans[N];
stack<int> c[N],st;
bool vis[N];
int dfs(int u){
	if(ans[u]) return ans[u];
	if(vis[u])
		while(1){
			int v=st.top(); st.pop();
			vis[v]=0;
			c[v].pop();
			if(v==u)
			break;
		}
	vis[u]=1;
	st.push(u);
	if(c[u].empty()) 
	return u;
	return dfs(c[u].top());
}
int main(){
	freopen("stack.in","r",stdin);
	freopen("stack.out","w",stdout);
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>k[i];
		for(int j=1;j<=k[i];j++){
			int c1;
			cin>>c1;
			c[i].push(c1);
		}
	}
	for(int i=1;i<=n;i++){
		if(!ans[i]){
			ans[i]=dfs(i);
			while(!st.empty()){
				ans[st.top()]=ans[i];
				st.pop();
			}
		}
		cout<<ans[i]<<" ";
	}
	return 0;
}