POJ 3050：Hopscotch（dfs + set容器）

原题地址：点击打开链接

题目大致意思：

1.5*5的方阵中，先随意挑一格，可以上下左右走，走5次，每走一次记录下所走格子的数字，把总共6个数字连起来，形成一串数字。求总共可以形成多少种不同的数字串

代码：

#include <iostream>
#include <set>

using namespace std;

int m[5][5];
int dx[] = {0, 0, 1, -1}, dy[] = {1, -1, 0, 0};
set<int> flag;


void dfs(int x, int y, int k, int sum)
{
	if(6 == k){
		flag.insert(sum);
		return ;
	}
	
	for(int i = 0; i < 4; i ++){
		int nx = x + dx[i], ny = y + dy[i];
		if(nx >= 0 && nx < 5 && ny >= 0 && ny < 5)
			dfs(nx, ny, k + 1, sum * 10 + m[nx][ny]);
	}
	
}

void solve()
{
	for(int i = 0; i < 5; i ++)
		for(int j = 0; j < 5; j ++)
			dfs(i, j, 1, m[i][j]);
	cout << flag.size() << endl;
} 

int main()
{
	for(int i = 0; i < 5; i ++)
		for(int j = 0; j < 5; j ++)
			cin >> m[i][j];
	solve();
	return 0;
}

因为题目只是5*5的格子，穷举也才25*4^5 = 25600种，然后用set容器记录就可以了，而set容器可以自动舍弃重复的，因此快速得解！

之前用的是一个长度为1000000的数组标记数字是否访问过，而使用set容器就大大地节省了空间