错题集7——池塘里的鱼

题目信息

解题思路

算法1（对于前40%数据）：DP+滚动数组优化

• **状态定义
  ** dp_i为还剩i条鱼时第一条鱼的存活概率。
• 滚动数组：
  
• 时间复杂度: O(nlogn2³²)

算法2（对于后60%数据）：状压DP

• **状态定义：
  ** dp_k为到达状态k的概率。
• 状态转移：
  
• tips: 上述popcnt为库函数 __builtin_popcount，求一个数的二进制表示中有多少个1。

代码实现

#include <bits/stdc++.h>
using namespace std;
const int N = 17;
int n, p[1 << N];
double a[N][N], dp[1 << N];
bool check(int x) {
	if (n > 15) {
		dp[1] = 1;
		for (int i = 2; i <= n; ++i) {
			dp[i] = dp[i - 1] * ((double) (i - 2) / i + 2. / i * x / (x + 1));
		}
		return dp[n] >= 0.5;
	}
	memset(dp, 0, sizeof dp);
	p[1] = x;
    for (int i = 1; i <= n; ++i) {
        for (int j = 1; j <= n; ++j) {
            a[i][j] = (double) p[i] / (p[i] + p[j]);
        }
    }
	dp[(1 << n) - 1] = 1;
    for (int k = (1 << n) - 1; k; --k) {
        int t = __builtin_popcount(k);
        t = t * (t - 1) / 2;
        for (int i = 1; i <= n; ++i) {
            if (k >> (i - 1) & 1) {
                for (int j = 1; j <= n; ++j) {
                    if (i != j && k >> (j - 1) & 1) {
                        dp[k ^ (1 << (j - 1))] += dp[k] * a[i][j] / t;
                    }
                }
            }
        }
    }
	return dp[1] >= 0.5;
}
int main() { 
	cin >> n;
	for (int i = 2; i <= n; ++i) {
        cin >> p[i];
    }
	int l = 1, r = 1e5, res = -1;
	while (l <= r) {
		int mid = l + r >> 1;
		if (check(mid)) {
			res = mid;
            r = mid - 1;
        } else {
			l = mid + 1;
        }
	}
	cout << res << '\n';
	return 0;
}