BZOJ4117 : [Wf2015]Weather Report

时间:2023-03-08 20:21:15

一种天气情况的概率只与4种天气的出现次数有关,故将相同概率的情况计数后放入堆中模拟哈夫曼树即可。

每次取出概率最小的,将它个数除以2,对于零头需要特判。

#include<cstdio>
#include<queue>
#define N 22
using namespace std;
typedef long long ll;
typedef pair<double,ll>P;
int n,i,j,k,l;ll c[N][N];double A,B,C,D,pa[N],pb[N],pc[N],pd[N],ans;
priority_queue<P,vector<P>,greater<P> >q;
int main(){
scanf("%d%lf%lf%lf%lf",&n,&A,&B,&C,&D);
for(c[0][0]=i=1;i<=n;i++)for(c[i][0]=j=1;j<=i;j++)c[i][j]=c[i-1][j-1]+c[i-1][j];
for(pa[0]=pb[0]=pc[0]=pd[0]=1,i=1;i<=n;i++){
pa[i]=pa[i-1]*A;
pb[i]=pb[i-1]*B;
pc[i]=pc[i-1]*C;
pd[i]=pd[i-1]*D;
}
for(i=0;i<=n;i++)for(j=0;j<=n;j++)for(k=0;k<=n;k++)for(l=0;l<=n;l++)
if(i+j+k+l==n)
q.push(P(pa[i]*pb[j]*pc[k]*pd[l],c[n][i]*c[n-i][j]*c[n-i-j][k]));
while(1){
P t=q.top();q.pop();
if(q.empty()&&t.second==1)break;
if(t.second>1){
if(t.second&1)q.push(P(t.first,1)),t.second--;
ans+=t.first*t.second;
q.push(P(t.first*2,t.second>>1));
}else{
P u=q.top();q.pop();
ans+=t.first+u.first;
q.push(P(t.first+u.first,1));
if(u.second>1)q.push(P(u.first,u.second-1));
}
}
return printf("%.6f",ans),0;
}