【和小安一起刷题】POJ3050 Hopscotch ----深度优先搜索（DFS）

01、题目描述

小伙伴们大家好，这里是和 【和小安一起刷题 】 系列栏目，今天我们要做的依旧是一道搜索题——POJ3050 Hopscotch，今天要用到的知识有，深度优先搜索（DFS） 以及 C++< set >库（set中不允许有相同元素）的一些使用。

题目链接： POJ3050 Hopscotch.
题目描述：输入一个5x5的整数矩阵，从任意数字开始，可以向前后左右四个方向运动，运动5次后，得到一个由6个数字组成的序列，问共有多少种不同的序列。
示例输入：

示例输出： 15（共有15种不同的序列，111111, 111112, 111121, 111211, 111212, 112111, 112121, 121111, 121112, 121211, 121212, 211111, 211121, 212111, 212121）

02、解题思路

看完题目后，很自然的想到，如果能算出所有的序列，然后删去重复的序列，就可以得到答案，同时因为本题数据量不大，所以可以不作其他处理。通过DFS深搜的方法，从一个位置开始，运动至得到一个6个数的序列，然后返回上一步（既只有5个数的状态），再向另外的方向运动，又返回，又运动，如此反复就可以得到所有的序列。然后删去重复的序列，只需要把DFS搜索出的所有序列insert到set中去，就可以删去重复序列，最后再 set.size() 就可以得到答案。

03、题解代码

#include <iostream>
#include <set>
using namespace std;

int map[6][6];
set<int> ans;		
int dx[4] = {-1, 1, 0, 0};
int dy[4] = {0, 0, 1, -1};

void dfs(int x, int y, int step, int temp) {
	if(step == 5) {
		ans.insert(temp);
		return ;
	}
	
	for(int i = 0; i < 4; i++) {
		int nx = dx[i] + x;
		int ny = dy[i] + y;	
		if(nx>=0 && nx<5 && ny>=0 && ny<5){
			dfs(nx, ny, step+1, temp*10+map[nx][ny]);
		}
	}
}

int main() {
	for(int i = 0; i < 5; i++)
		for(int j = 0; j < 5; j++)
			cin >> map[i][j];
	
	for(int i = 0; i < 5; i++)
		for(int j = 0; j < 5; j++)
			dfs(i, j, 0, map[i][j]);
	
	cout << ans.size() << endl;
	
	return 0;
}

让我们一起向更优秀的自己前进。