cf B. Resort

http://codeforces.com/contest/350/problem/B

从旅馆开始倒着找到一个点它的出度>1的位置为止，比较长度大小，找到一个长度最大的即可。

#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 100010
using namespace std;

int n;
int a[maxn],b[maxn];
int out[maxn];
int k;

void print(int x)
{
    if(b[x]&&out[b[x]]==1)
    {
        print(b[x]);
        printf(" %d",x);
    }
    else
    {
        printf("%d",x);
        return ;
    }
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&a[i]);
        }
        for(int j=1; j<=n; j++)
        {
            scanf("%d",&b[j]);
            out[b[j]]++;
        }
        int max1=0;
        for(int i=1; i<=n; i++)
        {
            if(a[i])
            {
                int t1=1;
                for(int j=i; b[j]!=0&&out[b[j]]==1; j=b[j])
                {
                    t1++;
                }
                if(max1<t1)
                {
                    max1=t1;
                    k=i;
                }
            }
        }
        printf("%d
",max1);
        print(k);
        printf("
");
    }
    return 0;
}