Acjoy群赛02-C - HopscotcH

转载于 2015-12-21 16:58:00 发布 · 92 阅读

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/wlxtuacm/p/5712294.html

本文介绍了一个基于深度优先搜索(DFS)的算法，用于在一个5x5的矩阵中寻找不重复的六位数组合。该算法使用了map来记录已经出现过的组合，并通过递归调用DFS遍历所有可能的路径。

#include <stdio.h>
#include <string.h>
#include <map>

int arr[5][5], ans;
const int dir[4][2] = {0, 1, 0, -1, 1, 0, -1, 0};
std::map<int, int> mp;//利用map查重，本来是用set的，然后超时233.。。

void DFS(int x, int y, int k, int sum) {
    if(k == 6) {
        if(mp[sum]++ == 0) ++ans;
        return;
    }
    for(int i = 0, a, b; i < 4; ++i) {
        a = x+dir[i][0];
        b = y+dir[i][1];
        if(a>=0&&b>=0&&a<5&&b<5)
            DFS(a, b, k+1, sum*10+arr[x][y]);
    }
}

int main() {
    int i, j;
    for(i = 0; i < 5; ++i)
        for(j = 0; j < 5; ++j)
            scanf("%d", &arr[i][j]);
    for(i = 0; i < 5; ++i)
        for(j = 0; j < 5; ++j)
            DFS(i, j, 0, 0);
    printf("%d\n", ans);
    return 0;
}

转载于:https://www.cnblogs.com/wlxtuacm/p/5712294.html