题意
一个矩阵,每个元素标有方向,人可以从任意一个位置出发,如果该位置永远走不到边缘,则被认为被困住了,统计这种位置的个数。
矩阵有?
,其能代表某一种方向。
比如
3 3
?U?
R?L
RDL
答案是5
解决方案
用DFS暴搜
对于普通的节点(非?
),搜索如果能走出去,则回溯0,并且做类似并查集的那种操作,把这条路径都标记为0,同理,如果走不出去或者遇到?
,则标记为1。
但是会遇到该情况,当前DFS走过的路径如果再走到,意味着出现环。但是如果遇到了之前走过的路径(非本次DFS搜索到的),就直接利用该路径的结论。
但是这两种情况代码写起来会混淆,所以用了一点细节,每次走的时候都默认给其标记为是1,最后回溯回来如果能走出,再更新为0。
对于?
节点,他考虑8种条件,其上下左右存在?
,或者上下左右存在走不出去的路径节点。
代码
挺考验码力的一道题
#include<bits/stdc++.h>
#include <vector>
using namespace std;
typedef long long ll;
void solve()
{
int n, m;
cin >> n >> m;
vector<vector<char>> g(n + 2, vector<char>(m + 2));
for (int i = 1; i <= n; i++)
{
for (int j = 1; j <= m; j++)
{
cin >> g[i][j];
}
}
vector<vector<int>> vis(n + 2, vector<int>(m + 2));
vector<vector<int>> die(n + 2, vector<int>(m + 2));
int cnt = 0;
auto dfs = [&] (auto self, int i, int j) -> bool
{
vis[i][j] = 1;
if (g[i][j] == '?')
return die[i][j] = 1;
int ni = i, nj = j;
if (g[i][j] == 'U') ni--;
if (g[i][j] == 'D') ni++;
if (g[i][j] == 'L') nj--;
if (g[i][j] == 'R') nj++;
if (ni < 1 || ni > n || nj < 1 || nj > m) return die[i][j] = 0;
if (vis[ni][nj]) return die[i][j] = die[ni][nj]; //上一轮访问过,直接用结果。 或者是环
die[i][j] = 1;
return die[i][j] = self(self, ni, nj);
};
for (int i = 1; i <= n; i++)
{
for (int j = 1; j <= m; j++)
{
if (!vis[i][j] && g[i][j] != '?')
{
dfs(dfs, i, j);
}
}
}
for (int i = 1; i <= n; i++)
{
for (int j = 1; j <= m; j++)
{
if (g[i][j] == '?' && (die[i + 1][j] || die[i - 1][j] || die[i][j + 1] || die[i][j - 1] || g[i + 1][j] == '?' || g[i - 1][j] == '?' || g[i][j + 1] == '?' || g[i][j - 1] == '?'))
cnt++;
else
cnt += die[i][j];
}
}
cout << cnt << endl;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
int t;
cin >> t;
while (t--)
{
solve();
}
}