zzulioj--1831-- 周末出游(vector建图+dfs)

1831: 周末出游

Time Limit: 1 Sec Memory Limit:
128 MB

Submit: 22 Solved: 8

SubmitStatusWeb
Board

Description

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

Input

每组第一行输入n，表示有n个人。然后第二行输入n个数a1，a2，a3.....(0<ai<=n)表示第i个人的希望自己的那位好朋友要去。如果ai=i表示这位同学一定会去（注孤）。(0<n<1000)

Output

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

Sample Input

41 2 3 3

Sample Output

yes

HINT

刚开始想的太麻烦了，以为还可以邀请一个人，后来才发现折磨简单

#include<stdio.h>
#include<string.h>
#include<vector>
#include<algorithm>
using namespace std;
vector<int>g[10010];
int vis[10010];
int ans;
void dfs(int u)
{
for(int i=0;i<g[u].size();i++)
{
ans++;
dfs(g[u][i]);
}
}
int main()
{
int n;
while(scanf("%d",&n)!=EOF)
{
memset(vis,0,sizeof(vis));
ans=0;
int a;
for(int i=0;i<=n;i++)
g[i].clear();
for(int i=1;i<=n;i++)
{
scanf("%d",&a);
if(a==i)
{
vis[a]=1;
}
else
g[a].push_back(i);
}
for(int i=1;i<=n;i++)
if(vis[i])
{
ans++;
dfs(i);
}

if(ans==n)
printf("yes\n");
else
printf("no\n");
}
return 0;
}