BZOJ4117 : [Wf2015]Weather Report

一种天气情况的概率只与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;
}