loj#6072 苹果树(折半搜索,矩阵树定理,容斥)
题解时间
$ n \le 40 $ 。
无比精确的数字。
很明显只要一个方案不超过 $ limits $ ,之后的计算就跟选哪个没关系了。
折半搜索排序来统计有i个果子是有用的情况下的方案数。
然后矩阵树求生成树个数,容斥乱搞。
#include<bits/stdc++.h>
using namespace std;
template<typename TP>inline void read(TP &tar)
{
TP ret=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){ret=ret*10+(ch-'0');ch=getchar();}
tar=ret*f;
}
namespace RKK
{
const int N=50,mo=1000000007;
void doadd(int &a,int b){if((a+=b)>=mo)a-=mo;}
int add(int a,int b){return (a+=b)>=mo?a-mo:a;}
int fpow(int a,int p){int ret=1;while(p){if(p&1)ret=1ll*ret*a%mo;a=1ll*a*a%mo,p>>=1;}return ret;}
struct pat{int x,y;bool operator < (const pat &p)const{return x<p.x;}};
int n,halfn,lim,val[N],msn;
pat l1[1145141];int tp1;
pat l2[1145141];int tp2;
void dfs1(int x=1,int sum=0,int cnt=0)
{
if(sum>lim) return;if(x>halfn){l1[++tp1]=(pat){sum,cnt};return;}
dfs1(x+1,sum,cnt);if(~val[x]) dfs1(x+1,sum+val[x],cnt+1);
}
void dfs2(int x=halfn+1,int sum=0,int cnt=0)
{
if(sum>lim) return;if(x>n){l2[++tp2]=(pat){sum,cnt};return;}
dfs2(x+1,sum,cnt);if(~val[x]) dfs2(x+1,sum+val[x],cnt+1);
}
int c[N][N];void init(){for(int i=0;i<=40;i++){c[i][0]=1;for(int j=1;j<=i;j++)c[i][j]=add(c[i-1][j-1],c[i-1][j]);}}
int ma[N][N];
int calc(int sn)
{
memset(ma,0,sizeof(ma));
for(int i=1;i<=n;i++)for(int j=i+1;j<=n;j++)
{
if((i<=sn&&(j<=sn||j>msn))||i>msn||j>msn)
ma[i][i]++,ma[j][j]++,doadd(ma[i][j],mo-1),doadd(ma[j][i],mo-1);
}
int b=n-1;int f=1;
for(int l=1;l<=b;l++)
{
int g=l;for(;g<=b&&!ma[g][l];g++);if(g>b) return 0;
if(g!=l){for(int j=l;j<=b;j++) swap(ma[l][j],ma[g][j]);f=-f;}
for(g=l+1;g<=b;g++)
{
int k=1ll*ma[g][l]*fpow(ma[l][l],mo-2)%mo;
for(int j=l;j<=b;j++) doadd(ma[g][j],mo-1ll*ma[l][j]*k%mo);
}
}
if(f==-1) f=mo-1;
for(int i=1;i<=b;i++) f=1ll*f*ma[i][i]%mo;
return f;
}
int cnt[N],sum[N];
int cnttmp[N];
int main()
{
read(n),read(lim),halfn=n+1>>1;
for(int i=1;i<=n;i++) read(val[i]),msn+=(val[i]!=-1);
dfs1(),dfs2();init();
sort(l1+1,l1+tp1+1),sort(l2+1,l2+tp2+1);
for(int i1=tp1,i2=1;i1;i1--)
{
for(;i2<=tp2&&l1[i1].x+l2[i2].x<=lim;cnttmp[l2[i2].y]++,i2++);
for(int j=0;j<=n;j++) doadd(cnt[l1[i1].y+j],cnttmp[j]);
}
for(int i=0;i<=msn;i++) sum[i]=calc(i);
for(int i=1;i<=msn;i++)for(int j=0;j<i;j++)
doadd(sum[i],mo-1ll*c[i][j]*sum[j]%mo);
int ans=0;
for(int i=0;i<=msn;i++) doadd(ans,1ll*cnt[i]*sum[i]%mo);
printf("%d\n",ans);
return 0;
}
}
int main(){return RKK::main();}