ZUFE 问题 D: 周末出游

题目描述

周末天气真好，大家组织一起出去玩（玩你妹，不好好学习），可是，有些人要知道（有好朋友了不起呀）自己最要好的朋友接受邀请了，他才会去，只有邀请到他们最好的朋友才会去（贱人就是矫情）。可是作为负责人的你（冤大头）必须判断是否能够办好这次聚会（气死爸爸拉，哼），邀请到所有的人。

输入

输入包含多组测试数据，每组测试数据第一行输入n（0<n<1000），表示有n个人。然后第二行输入n个数，a[1]，a[2]，a[3].....a[n] (0<a[i]<=n) 。

对于第 i 个人来说，a[i] 是他的好朋友， 如果 a[i] 参加了，那么 i 也会参加。

如果a[i]=i，表示编号为 i 的同学一定会去（注孤）。

输出

如果能邀请到所有输出yes，反之no。

样例输入

4
1 2 3 3

样例输出

yes

提示

对样例数据的解释：第一行输入了4，表示有4个人，编号分别为1,2,3,4。

第二行，

a[1]=1表示编号为1的同学一定会去；

a[2]=2表示编号为2的同学一定会去；

a[3]=3表示编号为3的同学一定会去；

a[4]=3表示如果编号为3的同学去了，那么编号为4的同学也会去。

因为已经确定3一定去，所以4也去了。

因此所有人都去了，输出yes。

数据很小，暴力过

#include<bits/stdc++.h>
 
using namespace std;
 
int f[20010],dp[20010],  n, m, k, sum = 0;
 
int getf(int v)
{
    if(f[v] == v)
        return v;
    else
    {
        f[v] = getf(f[v]);
        return f[v];
    }
}
 
void merge(int x,int y)
{
    int t1, t2;
    t1 = getf(x);
    t2 = getf(y);
    if(t1 != t2)
    {
        f[t2] = t1;
    }
    return ;
}
 
int main()
{
    int a , b;
    while(cin >> n)
    {
        memset(dp, 0 , sizeof(dp));
        memset(f,0,sizeof(f));
        for(int i = 1; i <= n ; i ++)
        {
            cin >> f[i];
            if(f[i] == i)
                dp[i] = 1;
        }
        for(int j = 1 ; j <= 2000 ; j ++)
        {
            for(int i = 1; i <= n ; i ++)
            {
                if(dp[f[i]] == 1)
                    dp[i] = 1;
            }
        }
        int flag = 1;
        for(int i = 1; i <= n ; i ++)
        {
            if(dp[i] == 0)
            {
                flag = 0;
                break;
            }
        }
        if(flag)
            cout << "yes" << endl;
        else
            cout << "no" << endl;
    }
    return 0;
}