0828-T4 聰聰與可可

maniubi發表於2024-08-29

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

相關文章