zzulioj 1831: 周末出游 （vector&&dfs）

1831: 周末出游

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 29   Solved: 9

Submit Status Web Board

Description

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

Input

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

Output

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

Sample Input

4

1 2 3 3

Sample Output

yes

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<vector>
#define N 1010
using namespace std;
int vis[N];
vector<int>q[N];
int cnt;
int dfs(int n)
{
	for(int i=0;i<q[n].size();i++)
	{
		cnt++;
		dfs(q[n][i]);
			
	}
}
int main()
{
	int n,i,j,x;
	while(scanf("%d",&n)!=EOF)
	{
		memset(vis,0,sizeof(vis));
		for(i=0;i<=n;i++)	
			q[i].clear();
		for(i=1;i<=n;i++)
		{
			scanf("%d",&x);
			if(x==i)
				vis[i]=1;
			else
				q[x].push_back(i);
		}
		cnt=0;
		for(i=1;i<=n;i++)
		{
			if(vis[i])
			{
				cnt++;
				dfs(i);
			}
		}
		if(cnt==n)
			printf("yes\n");
		else
			printf("no\n");
	}
	return 0;
}

//用邻接表解

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 1010
using namespace std;
struct zz
{
	int from;
	int to;
	int step;
	int next;
}edge[N<<1];
int head[N],top;
int vis[N];
int cnt;
void add(int u,int v)
{
	edge[top].from=u;
	edge[top].to=v;
	edge[top].next=head[u];
	head[u]=top++;
}
void dfs(int n)
{
	int i,j,k;
	if(head[n]==-1)
		return ;
	for(i=head[n];i!=-1;i=edge[i].next)
	{
		cnt++;
		k=edge[i].to;
		dfs(k);
	}
}
int main()
{
	int n,i,j;
	int x;
	while(scanf("%d",&n)!=EOF)
	{
		memset(vis,0,sizeof(vis));
		memset(head,-1,sizeof(head));
		top=0;
		for(i=1;i<=n;i++)
		{
			scanf("%d",&x);
			if(x==i)
				vis[i]=1;
			else
				add(x,i);
		}
		cnt=0;
		for(i=1;i<=n;i++)
		{
			if(vis[i])
			{
				cnt++;
				dfs(i);
			}
		}
		if(cnt==n)
			printf("yes\n");
		else
			printf("no\n");
	}
	return 0;
}