Div-2 E 求调。
  • 板块学术版
  • 楼主Demeanor_Roy
  • 当前回复3
  • 已保存回复3
  • 发布时间2023/10/8 19:38
  • 上次更新2023/11/2 14:54:03
查看原帖
Div-2 E 求调。
297806
Demeanor_Roy楼主2023/10/8 19:38
#include<bits/stdc++.h>
using namespace std;

const int N=2e5+10;
int n,top,f[N],to[N],fa[N],stk[N];
vector<int> b[N],cur[N],vec[N];
bool vis[N],iscur[N];
inline int get(int x){return x==fa[x]?x:fa[x]=get(fa[x]);}
inline void dfs(int u)
{
	for(auto v:vec[u]) 
	{
		if(iscur[v]) continue;
		dfs(v);
		if(!f[v]) f[u]=1;
	}
	if(iscur[u]&&!f[u]) f[u]=-1;
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++) fa[i]=i;
	for(int i=1;i<=n;i++) 
	{
		scanf("%d",&to[i]);
		fa[get(i)]=get(to[i]);
		vec[to[i]].emplace_back(i);
		if(to[i]==i) return puts("-1"),0;
	}
	for(int i=1;i<=n;i++) b[get(i)].push_back(i);
	for(int x=1;x<=n;x++)
	{
		if(!b[x].size()) continue;
		int now=b[x][0];
		do
		{
			vis[now]=true;
			stk[++top]=now;
			if(vis[to[now]])
			{
				do
				{
					iscur[stk[top]]=true;
					cur[x].emplace_back(stk[top]);
				}while(stk[top--]!=to[now]);
				break;
			}
			now=to[now];
		}while(true);
		for(auto v:cur[x]) dfs(v);	
		for(int i=0;i<(int)cur[x].size();i++)
		{
			if(~f[cur[x][i]]) continue;
			top=0;
			for(int j=i;j<(int)cur[x].size();j++)
			{
				if(~f[cur[x][j]]) break;			
				stk[++top]=cur[x][j];
				if(j==(int)cur[x].size()-1)
					for(int k=0;k<i;k++)
					{
						if(~f[cur[x][k]]) break;
						else stk[++top]=cur[x][k];
					}				
			}
			if(!i)
			{
				reverse(stk+1,stk+top+1);int lim=top;
				for(int j=(int)cur[x].size()-1;j>lim-1;j--)
				{
					if(~f[cur[x][j]]) break;			
					stk[++top]=cur[x][j];
				}
				reverse(stk+1,stk+top+1);
			}
			if(top&1) for(int i=1;i<=top;i++) f[stk[i]]=(i&1?0:1);
			else for(int i=1;i<=top;i++) f[stk[i]]=(i&1?1:0);
		}
	}
	int cnt=0;
	for(int i=1;i<=n;i++) if(!f[i]) ++cnt;
	printf("%d\n",cnt);
	for(int i=1;i<=n;i++) if(!f[i]) printf("%d ",to[i]);
	return 0;
}

第五个点 wa 了。求求了。

2023/10/8 19:38
加载中...