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

  }

}

  

原文地址:https://www.cnblogs.com/kiuhghcsc/p/5604829.html