原题地址:点击打开链接
题目大致意思:
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容器就大大地节省了空间