[NOIP2015 提高组] 信息传递
题目背景
NOIP2015 Day1T2
题目描述
有 n n n 个同学(编号为 1 1 1 到 n n n)正在玩一个信息传递的游戏。在游戏里每人都有一个固定的信息传递对象,其中,编号为 i i i 的同学的信息传递对象是编号为 T i T_i Ti 的同学。
游戏开始时,每人都只知道自己的生日。之后每一轮中,所有人会同时将自己当前所知的生日信息告诉各自的信息传递对象(注意:可能有人可以从若干人那里获取信息,但是每人只会把信息告诉一个人,即自己的信息传递对象)。当有人从别人口中得知自己的生日时,游戏结束。请问该游戏一共可以进行几轮?
输入格式
输入共 2 2 2 行。
第一行包含 1 1 1 个正整数 n n n,表示 n n n 个人。
第二行包含 n n n 个用空格隔开的正整数 T 1 , T 2 , ⋯ , T n T_1,T_2,\cdots,T_n T1,T2,⋯,Tn,其中第 i i i 个整数 T i T_i Ti 表示编号为 i i i 的同学的信息传递对象是编号为 T i T_i Ti 的同学, T i ≤ n T_i\leq n Ti≤n 且 T i ≠ i T_i\neq i Ti=i。
输出格式
共一行一个整数,表示游戏一共可以进行多少轮。
样例 #1
样例输入 #1
5
2 4 2 3 1
样例输出 #1
3
提示
样例 1 解释:
游戏的流程如图所示。当进行完第 3 3 3 轮游戏后, 4 4 4 号玩家会听到 2 2 2 号玩家告诉他自己的生日,所以答案为 3 3 3。当然,第 3 3 3 轮游戏后, 2 2 2 号玩家、 3 3 3 号玩家都能从自己的消息来源得知自己的生日,同样符合游戏结束的条件。
- 对于 30 % 30\% 30% 的数据, n ≤ 200 n\le 200 n≤200;
- 对于 60 % 60\% 60% 的数据, n ≤ 2500 n\le 2500 n≤2500;
- 对于 100 % 100\% 100% 的数据, n ≤ 2 × 1 0 5 n\le 2\times 10^5 n≤2×105。
代码1(拓扑序+dfs)
int n;
int w[N];
vector<int>G[N];
int din[N];
bool vis[N];
queue<int>q;
int sum;
void dfs(int u,int fa){
sum++;
for(auto& j:G[u]){
if(j==fa)continue;
vis[j]=1;
dfs(j,fa);
}
}
void solve(){
cin>>n;
for(int i=1;i<=n;i++){
cin>>w[i];
G[i].emplace_back(w[i]);
din[w[i]]++;
}
for(int i=1;i<=n;i++){
if(!din[i]){
q.push(i);
vis[i]=1;
}
}
while(q.size()){
int t=q.front();
q.pop();
for(auto& j:G[t]){
if(!(--din[j])){
vis[j]=1;
q.push(j);
}
}
}
int ans=1e18;
for(int i=1;i<=n;i++){
if(!vis[i]){
sum=0;
dfs(i,i);
ans=min(ans,sum);
}
}
cout<<ans<<endl;
}
核心代码(并查集)
int n,w[N];
int p[N];
int ans,cnt;
int find(int x,int& cnt){
cnt++;
if(x!=p[x])return find(p[x],cnt);//并查集这么写可以算出环的大小的
return p[x];
}
void solve(){
cin>>n;
for(int i=1;i<=n;i++)cin>>w[i],p[i]=i;
ans=1e18;
for(int i=1;i<=n;i++){
cnt=0;
if(find(w[i],cnt)==i){
ans=min(ans,cnt);
}else{
p[i]=w[i];
}
}
cout<<ans;
}
核心代码(targin+缩点)
int n,w[N];
int low[N],dfn[N];
int tmm,stk[N],top,scc_cnt;
int id[N],scc_siz[N];
bool in_stk[N];
vector<int>G[N];
void targin(int u){
low[u]=dfn[u]=++tmm;
stk[++top]=u,in_stk[u]=1;
for(auto& j:G[u]){
if(!dfn[j]){
targin(j);
low[u]=min(low[u],low[j]);
}else if(in_stk[j]){
low[u]=min(low[u],dfn[j]);
}
}
if(low[u]==dfn[u]){
scc_cnt++;
int y;
do{
y=stk[top--];
in_stk[y]=false;
id[y]=scc_cnt;
scc_siz[scc_cnt]++;
}while(y!=u);
}
}
void solve(){
cin>>n;
for(int i=1;i<=n;i++)cin>>w[i],G[i].emplace_back(w[i]);
for(int i=1;i<=n;i++){
if(!dfn[i]){
targin(i);
}
}
int ans=1e18;
for(int i=1;i<=scc_cnt;i++){
if(scc_siz[i]!=1)
ans=min(ans,scc_siz[i]);
}
if(ans==1e18)ans=-1;
cout<<ans;
}