hdu 1181 变形课 dfs

变形课

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 131072/65536 K (Java/Others)
Total Submission(s): 26454    Accepted Submission(s): 9668

Problem Description

呃......变形课上Harry碰到了一点小麻烦,因为他并不像Hermione那样能够记住所有的咒语而随意的将一个棒球变成刺猬什么的,但是他发现了变形咒语的一个统一规律:如果咒语是以a开头b结尾的一个单词,那么它的作用就恰好是使A物体变成B物体.
Harry已经将他所会的所有咒语都列成了一个表,他想让你帮忙计算一下他是否能完成老师的作业,将一个B(ball)变成一个M(Mouse),你知道,如果他自己不能完成的话,他就只好向Hermione请教,并且被迫听一大堆好好学习的道理.

 

Input

测试数据有多组。每组有多行，每行一个单词,仅包括小写字母,是Harry所会的所有咒语.数字0表示一组输入结束.

 

Output

如果Harry可以完成他的作业,就输出"Yes.",否则就输出"No."(不要忽略了句号)

 

Sample Input

so soon river goes them got moon begin big 0

 

Sample Output

Yes.

Hint

Hint

Harry 可以念这个咒语:"big-got-them".

#include
#include
#include
#include
using namespace std;
char a[1010][20];
int b[1010];
bool flag = false;
bool visit[1010];
int n;

void dfs(int index) {
	if (flag) {
		return;
	}
	if (a[index][strlen(a[index]) - 1] == 'm') {
		flag = true;
		return;
	}
	visit[index] = true;
	for (int i = 0; i < n; i++) {
		if (!visit[i] && a[i][0] == a[index][strlen(a[index]) - 1]) {
			visit[i] = true;
			dfs(i);
			visit[i] = false;
			if (flag)//剪枝操作
				return;
		}
	}
}

int main() {
	int i = 0;
	int j = 0;
	while (cin >> a[i]) {
		if (a[i][0] != '0') {
			if (a[i][0] == 'b') {
				b[j] = i;
				j++;
			}
			i++;
			continue;
		}
		n = i;
		for (int k = 0; k < j; k++) {
			fill(visit, visit + n, false);
			dfs(b[k]);
		}
		if (flag) {
			cout << "Yes." << endl;
		}
		else {
			cout << "No." << endl;
		}
		i = 0;
		j = 0;
		flag = false;
	}
}