Gifts by the List
题目链接:
http://codeforces.com/contest/681/problem/D
题意:
有n个人,他们之间有一些父子关系(自己也是自己的祖先),完了他们想要送礼,编号为i的人只想送给编号为a[i]的人,要求制作一张表,满足所有人在表上看到的第一个祖先就是他想要送礼的对象。
题解:
先dfs判断一下是否有满足的表并标记出出现在表上的编号:
当前节点为x,倒着往表里放点(祖先放在表的最下边),可以知道当a[x]是x的祖先且不是直接祖先(x和a[x]间还存在某点y使得y是x的祖先且a[x]是y的祖先)时不存在满足题意的表。
然后再将a[i]与i建边,拓扑排序输出答案即可
代码
#include<stdio.h>
#include<vector>
using namespace std;
const int N=100001;
vector<int>q[N];
bool mark[N],use[N],ans=true;
int a[N],res[N],sum=0,tail,in[N];
int mmin(int x,int y)
{
return x<y?x:y;
}
void dfs(int x,int fa)
{
if(!ans)return;
if(mark[a[x]]&&a[x]!=fa)
{
ans=false;
return;
}
use[a[x]]=true;
mark[x]=true;
for(int i=0;i<q[x].size();++i)
if(use[x])dfs(q[x][i],x);
else dfs(q[x][i],fa);
mark[a[x]]=false;
}
void tp()
{
for(int i=1;i<=tail;++i)
{
in[a[i]]=-1;
if(use[a[i]])res[++sum]=a[i];
for(int k=0;k<q[a[i]].size();++k)
{
int j=q[a[i]][k];
in[j]--;
if(in[j]==0)a[++tail]=j;
}
}
}
int main()
{
int n,m,x,y;
scanf("%d%d",&n,&m);
for(int i=0;i<=n;++i)
q[i].clear(),in[i]=res[i]=0,use[i]=mark[i]=false;
while(m--)
{
scanf("%d%d",&x,&y);
q[x].push_back(y);
in[y]++;
}
for(int i=1;i<=n;++i)
scanf("%d",&a[i]);
for(int i=1;i<=n;++i)
if(!in[i])use[a[i]]=true,dfs(i,a[i]);
if(!ans)printf("-1\n");
else
{
for(int i=1;i<=n;++i)
if(a[i]!=i)q[a[i]].push_back(i),in[i]++;for(int i=1;i<=n;++i)
if(!in[i])
{
a[tail=1]=i;
tp();
}
printf("%d\n",sum);
for(int i=sum;i>=1;--i)
printf("%d\n",res[i]);
}
}