题意
一个矩阵,每个元素标有方向,人可以从任意一个位置出发,如果该位置永远走不到边缘,则被认为被困住了,统计这种位置的个数。
矩阵有?
,其能代表某一种方向。
比如
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++;elsecnt += die[i][j];}}cout << cnt << endl;}int main(){ios::sync_with_stdio(false);cin.tie(0);int t;cin >> t;while (t--){solve();}}