Medium
Toss Strange Coins — C++
Full explanation · Time O(n^2) · Space O(n)
// Time: O(n^2)
// Space: O(n)
class Solution {
public:
double probabilityOfHeads(vector<double>& prob, int target) {
vector<double> dp(target + 1);
dp[0] = 1.0;
for (const auto& p : prob) {
for (int i = target; i >= 0; --i) {
dp[i] = ((i >= 1) ? dp[i - 1] : 0.0) * p + dp[i] * (1 - p);
}
}
return dp[target];
}
};