0828-T4 聰聰與可可
題意
貓抓老鼠。
貓每次會走到四周距離老鼠最近的點。若沒抓到老鼠還會再走一次。
老鼠每次會等機率向四周走一步,求貓抓到老鼠的期望時間。
思路
與處理出 \(nxt_{i,j}\) 表示貓在 \(i\) 老鼠在 \(j\),貓下一步走到哪裡。
\(f_{i,j}\) 表示貓在 \(i\) 老鼠在 \(j\),貓抓到老鼠的期望次數。
若 \(i=j\),\(f_{i,j} = 0\)。
若貓一步吃到老鼠,\(f_{i,j}=1\)。
\[f_{i,j}=\frac{1}{p}\sum f_{k,l} + 1
\]
\(k\) 表示貓下兩步走到哪裡,\(l\) 表示老鼠周圍的點,\(p\) 是老鼠周圍的點的個數加一。
使用記憶化搜尋實現。
程式碼
#include <bits/stdc++.h>
using namespace std;
const int N = 1005;
int n, m, nxt[N][N], d[N][N];
vector <int> e[N];
double f[N][N];
void bfs(int s) {
queue <int> q;
q.push(s);
while (!q.empty()) {
int p = q.front(); q.pop();
for (auto v : e[p]) {
if (d[s][v] || v == s) continue;
d[s][v] = d[s][p] + 1;
q.push(v);
}
}
}
double dfs(int cat, int mouse) {
if (cat == mouse) return f[cat][mouse] = 0;
int nxtcat = nxt[cat][mouse];
if (nxtcat == mouse) return f[cat][mouse] = 1;
nxtcat = nxt[nxtcat][mouse];
if (nxtcat == mouse) return f[cat][mouse] = 1;
if (f[cat][mouse]) return f[cat][mouse];
double res = 1, P = e[mouse].size() + 1;
res += dfs(nxtcat, mouse) / P;
for (auto nxtmouse : e[mouse])
res += dfs(nxtcat, nxtmouse) / P;
return f[cat][mouse] = res;
}
int main() {
cin >> n >> m;
int C, M; cin >> C >> M;
for (int i = 1, u, v; i <= m; i ++) {
cin >> u >> v;
e[u].push_back(v);
e[v].push_back(u);
}
for (int i = 1; i <= n; i ++) bfs(i);
for (int i = 1; i <= n; i ++) {
for (int j = 1; j <= n; j ++) {
d[0][j] = 1e9; int& p = nxt[i][j];
for (auto v : e[i]) {
if (d[v][j] < d[p][j]) p = v;
if (d[v][j] == d[p][j] && v < p) p = v;
}
}
}
dfs(C, M);
cout << fixed << setprecision(3) << f[C][M] << "\n";
return 0;
}