【AcWing 3502 每日一题·夏季】2021-05-13 暴搜

AcWing 3502

不同路径数

题目大意

给定一个 $n\times m$ 的二维矩阵，其中的每个元素都是一个 $[1,9]$ 之间的正整数。
从矩阵中的任意位置出发，每次可以沿上下左右四个方向前进一步，走过的位置可以重复走。
走了 $k$ 次后，经过的元素会构成一个 $(k+1)$ 位数。
请求出一共可以走出多少个不同的 $(k+1)$ 位数。

Time / Memory: 1 s / 64 MB

数据范围

对于$30\%$ 的数据, $1\le n,m\le 2,0\le k\le 2$
对于$100\%$ 的数据，$1\le n,m\le 5,0\le k\le 5,m\times n>1$

解题思路及分析

一开始还觉得挺难的，看了数据范围之后，就这？

数据范围很小，直接暴搜
注意可以重复走，不需要标记

AC代码

#include <bits/stdc++.h>
using namespace std;

const int N = 10;
int a[N][N];

int dx[] = {1, 0, -1, 0};
int dy[] = {0, 1, 0, -1};
int n, m, k;
set<int> ans;

bool judge(int xx, int yy)
{
    return 0 <= xx && xx < n && 0 <= yy && yy < m;
}

void dfs(int x, int y, int kk, int seq)
{
    if (kk > k)
    {
        ans.insert(seq);
        return ;
    }
    for (int i = 0; i < 4; i++)
    {
        int xx = x + dx[i];
        int yy = y + dy[i];
        if (judge(xx, yy)) 
        {
            dfs(xx, yy, kk + 1, seq * 10 + a[xx][yy]);
        }
    }
}

int main()
{
    scanf("%d%d%d", &n, &m, &k);
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < m; j++)
        {
            scanf("%d", &a[i][j]);
        }
    }
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < m; j++)
        {
            dfs(i, j, 0, 0);
        }
    }
    cout << ans.size() << endl;
    return 0;
}