Codeforces 681D：Gifts by the List (dfs+topsort)

　　

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 ");

　　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 ",sum);

　　　　for(int i=sum;i>=1;--i)

　　　　printf("%d ",res[i]);

　　}

}