题目描述
周末天气真好,大家组织一起出去玩(玩你妹,不好好学习),可是,有些人要知道(有好朋友了不起呀)自己最要好的朋友接受邀请了,他才会去,只有邀请到他们最好的朋友才会去(贱人就是矫情)。可是作为负责人的你(冤大头)必须判断是否能够办好这次聚会(气死爸爸拉,哼),邀请到所有的人。
输入
输入包含多组测试数据,每组测试数据第一行输入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; const int mn=1010; bool dp[mn]; int to[mn],fr[mn],nx[mn]; void dfs(int x) { dp[x]=1; //去 for(int i=fr[x];i!=-1;i=nx[i]) { int t=to[i]; //影响的人 if(!dp[t]) dfs(t); } } int main() { int n; while(~scanf("%d",&n)) { memset(dp,0,sizeof(dp)); memset(fr,-1,sizeof(fr)); int a[mn]; for(int i=1;i<=n;i++) { scanf("%d",&a[i]); to[i]=i; nx[i]=fr[a[i]]; fr[a[i]]=i; } //a[i]为起点 i为终点 for(int i=1;i<=n;i++) { if(i==a[i]&&!dp[i]) //必去的人 dfs(i); } bool flag=1; for(int i=1;i<=n;i++) { if(!dp[i]) flag=0; } if(flag) printf("yes\n"); else printf("no\n"); } return 0; }