POJ 3050 穷举

题意：给定一个5*5的地图，每个格子上有一个数字。从一个格子出发（上下左右4个方向），走5步将数字连起来可以构造出一个6位数。问该地图可以构造出多少个不同的6位数。

分析：可以对每个格子做深度优先遍历，构造出所有数字，但要注意不要重复计数。在这里，我使用了set来保存已构造出的数字，结果就是set中的元素个数。

#include <cstdio>
#include <set>
#include <algorithm>

using namespace std;

//输入
int a[5][5];

set<int> st;                //保存已构造的数字

const int dx[4] = {-1, 1, 0, 0};
const int dy[4] = {0, 0, -1, 1};

//深度优先遍历所有可能的构造
void dfs(int x, int y, int k, int num){
    if(k == 6){
        st.insert(num);
        return;
    }
    for(int i = 0; i < 4; i ++){
        int nx = x + dx[i], ny = y + dy[i];
        if(0 <= nx && nx < 5 && 0 <= ny && ny < 5){ 
            k ++;
            dfs(nx, ny, k, num * 10 + a[nx][ny]);
            k --;
        }
    }
}

void solve(){
    //对每个点作为起点，做深度优先遍历构造序列
    for(int i = 0; i < 5; i ++){
        for(int j = 0; j < 5; j ++){
            dfs(i, j, 1, a[i][j]);
        }
    }
    printf("%d\n", st.size());
}

int main(int argc, char const *argv[]){

    for(int i = 0; i < 5; i ++){
        for(int j = 0; j < 5; j ++)
            scanf("%d", &a[i][j]);
    }
    solve();

    return 0;
}