Acjoy群赛02-C - HopscotcH

#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;
}

你可能感兴趣的:(Acjoy群赛02-C - HopscotcH)