概率&期望DP
概率DP 求概率
Bag of mice
#include<bits/stdc++.h>
using namespace std;
const int N=1100;
double f[N][N];
int main(){
int w,b;
scanf("%d%d",&w,&b);
//f[0][0]=0;
for(int i=1;i<=w;i++) f[i][0]=1;
for(int i=1;i<=b;i++) f[0][i]=0;
for(int i=1;i<=w;i++){
for(int j=1;j<=b;j++){
f[i][j]+=(double)i/(i+j);
if(j-3>=0) f[i][j]+=(double)j/(i+j)*(j-1)/(i+j-1)*(j-2)/(i+j-2)*f[i][j-3];
if((i-1)>=0&&j-2>=0) f[i][j]+=(double)j/(i+j)*(j-1)/(i+j-1)*(i)/(i+j-2)*f[i-1][j-2];
}
}
printf("%.9lf",f[w][b]);
return 0;
}
Football
题意:有2^n支球队比赛,每次和相邻的球队踢,两两淘汰,给定任意两支球队相互踢赢的概率,求最后哪只球队最可能夺冠
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=130;
double f[8][N],p[N][N];
int main(){
int n;
while(scanf("%d",&n),n!=-1){
int m=1<<n;
for(int i=0;i<m;i++){
for(int j=0;j<m;j++)
scanf("%lf",&p[i][j]);
}
memset(f,0,sizeof f);
for(int i=0;i<m;i++) f[0][i]=1;
for(int i=1;i<=n;i++){
for(int j=0;j<m;j++){
for(int k=0;k<m;k++){
if(((j>>(i-1))^1)==(k>>(i-1))){//敲黑板 ==的优先级比^高,所有前面运算要加括号
f[i][j]+=f[i-1][j]*f[i-1][k]*p[j][k];
}
}
}
}
int ans;
double mx=0;
for(int i=0;i<m;i++){
if(f[n][i]>mx){
mx=f[n][i];
ans=i;
}
}
printf("%d\n",ans+1);
}
return 0;
}
多用括号是个好习惯,可以避免很多没必要且难调的bug
Jon and Orbs
分析:
#include<bits/stdc++.h>
using namespace std;
const int N=1e4+5,M=1e3+5;
double dp[N][M];
int K,Q;
int p[N],n,m=10000;
int main(){
cin>>n>>Q;
for(int i=1;i<=Q;i++){
scanf("%d",&p[i]);
}
dp[0][0]=1;
for(int i=1;i<=m;i++)
for(int j=1;j<=n;j++)
dp[i][j]=dp[i-1][j]*(double)j/n+dp[i-1][j-1]*(double)(n-j+1)/(n);
for(int i=1;i<=Q;i++){
int t=1;
for(int j=1;j<=m;j++) if(dp[j][n]<p[i]*0.0005) t++;
printf("%d\n",t);
}
// cout<<dp[m][n]<<endl;
system("pause");
return 0;
}
优化空间
#include<bits/stdc++.h>
using namespace std;
const int N=1e4+10;
const double eps=1e-8;
double f[N],ans[N];
int main(){
int n,q,p;
cin>>n>>q;
f[0]=1;//第0天没有物品的概率为1
for(int i=1;i<N;i++){
for(int j=n;j>0;j--){
f[j]=((double)j*f[j]+(n-j+1)*f[j-1])/n;
}
ans[i]=f[n];
f[0]=0;//第1、2、3...天物品种类为0的概率为0,必选了物品
}
while(q--){
cin>>p;
// cout<<"---- "<<p<<endl;
for(int i=1;i<N;i++){
if(ans[i]*2000>=p-eps){
cout<<i<<endl;
break;
}
}
}
return 0;
}
概率DP 求期望
Collecting Bugs
#include<iostream>
#include<cstring>
using namespace std;
int n,s;
double f[1010][1010];
int main() {
scanf("%d %d", &n, &s);
f[n][s]=0;
for(int i=n; i>=0; i--) {
for(int j=s; j>=0; j--) {
if(i==n && s==j) continue;
f[i][j]=(f[i][j+1]*i*(s-j)+f[i+1][j]*(n-i)*j+f[i+1][j+1]*(n-i)*(s-j)+n*s)/(n*s-i*j);
}
}
printf("%.4lf\n", f[0][0]);
}
换教室
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
const int N=2005;
int n,m,v,e,c[N],d[N];
double p[N],dis[N][N],f[N][N][2];
int main(){
cin>>n>>m>>v>>e;//时段数n,可换m节,教室数量v,道路数量e
for(int i=1;i<=n;i++) scanf("%d",&c[i]); //上课教室
for(int i=1;i<=n;i++) scanf("%d",&d[i]); //同课教室
for(int i=1;i<=n;i++) scanf("%lf",&p[i]);//通过概率
for(int i=1;i<=v;i++)
for(int j=1;j<i;j++)
dis[i][j]=dis[j][i]=2e9;
for(int i=1;i<=e;i++){
int a,b,c; scanf("%d%d%d",&a,&b,&c);
dis[a][b]=dis[b][a]=min(dis[a][b],1.0*c); //连边
}
for(int k=1;k<=v;k++) //Floyd求最短路
for(int i=1;i<=v;i++)
for(int j=1;j<i;j++)
if(dis[i][k]+dis[k][j]<dis[i][j])
dis[i][j]=dis[j][i]=dis[i][k]+dis[k][j];
for(int i=1;i<=n;i++)
for(int j=0;j<=m;j++)
f[i][j][0]=f[i][j][1]=2e9;
f[1][0][0]=f[1][1][1]=0;
for(int i=2;i<=n;i++){ //第i时段
for(int j=0;j<=min(m,i);j++){ //已用了j次机会
f[i][j][0]=min(f[i-1][j][0] //上次不换
+dis[c[i-1]][c[i]],
f[i-1][j][1] //上次申请换
+dis[c[i-1]][c[i]]*(1-p[i-1])
+dis[d[i-1]][c[i]]*p[i-1]);
if(j>0)
f[i][j][1]=min(f[i-1][j-1][0] //上次不换
+dis[c[i-1]][c[i]]*(1-p[i])
+dis[c[i-1]][d[i]]*p[i],
f[i-1][j-1][1] //上次申请换
+dis[c[i-1]][c[i]]*(1-p[i-1])*(1-p[i])
+dis[c[i-1]][d[i]]*(1-p[i-1])*p[i]
+dis[d[i-1]][c[i]]*p[i-1]*(1-p[i])
+dis[d[i-1]][d[i]]*p[i-1]*p[i]);
}
}
double ans=2e9;
for(int i=0;i<=m;i++)
ans=min(ans,min(f[n][i][0],f[n][i][1]));
printf("%.2lf",ans);
}
概率DP 求期望 拓扑排序(无环)
绿豆蛙的归宿
有两种方法求:记忆化搜索、拓扑排序
记忆化搜索
#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
int h[N],e[N],ne[N],w[N],idx;
int dout[N];
double f[N];
int n,m;
void add(int a,int b,int c){
e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
int dfs(int u){
if(f[u]) return f[u];
for(int i=h[u];~i;i=ne[i]){
int j=e[i];
dfs(j);
f[u]+=(f[j]+w[i])/dout[u];
}
return f[u];
}
int main(){
memset(h,-1,sizeof h);
cin>>n>>m;
int a,b,c;
while(m--){
cin>>a>>b>>c;
add(a,b,c);
dout[a]++;
}
f[n]=0;//double类型不能用memset
dfs(1);
printf("%.2lf",f[1]);
return 0;
}
拓扑排序(反向建图)
#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
int h[N],e[N],ne[N],w[N],idx;
int din[N],k[N];
double f[N];
int n,m;
void add(int a,int b,int c){
e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
void topsort(){
queue<int> q;
q.push(n);
while(q.size()){
int t=q.front();
q.pop();
for(int i=h[t];~i;i=ne[i]){
int j=e[i];
f[j]+=(f[t]+w[i])/din[j];
if(--k[j]==0) q.push(j);
}
}
}
int main(){
memset(h,-1,sizeof h);
cin>>n>>m;
int a,b,c;
while(m--){
cin>>a>>b>>c;
add(b,a,c);
din[a]++,k[a]++;
}
topsort();
printf("%.2lf",f[1]);
return 0;
}
小C的学习计划
#include<bits/stdc++.h>
using namespace std;
typedef long double ld;
typedef long long ll;
const int N=20;
ld f[1<<20];
int a[N];
int main(){
int n,m;
cin>>n>>m;
for(int i=0;i<n;i++) cin>>a[i];
for(int i=(1<<n)-1;~i;i--){
ld s=1;
ll sum=0,cnt=0;
for(int j=0;j<n;j++){
if((i>>j)&1) sum+=a[j],cnt++;
else s+=1.0/n*f[i|1<<j];
}
if(sum>=m) f[i]=0;
else{
f[i]=s/(1-(ld)cnt/n);
}
}
printf("%.6Lf",f[0]);
}
概率DP 求期望 高斯消元(有环)
学习链接
后面补上