题目描述
给定一个N列的表格,每列的高度各不相同,但底部对齐,然后向表格中填入K个相同的数,填写时要求不能有两个数在同一列,或同一行,下图中b是错误的填写,a是正确的填写,因为两个a虽然在同一行,但它们中间的表格断开。
输出所有填写方案数对1 000 000 007的余数。
输入: 第一行两个整数 N 和 K (1 ≤ N ≤ 500, 1 ≤ K ≤ 500),表示表格的列数和要填写的数的个数。 接下来一行N个数,表示每列的高度。高度不超过 1 000 000.
输出: 一个整数,方案总数对1000 000 007的余数。
题解
这种问题不好在序列上直接处理,考虑每次从当前序列中找出最小的数作为根,递归构造两边的子树。
这样构造出来的数叫笛卡尔树,保证了中序遍历是原序列,整颗树又满足堆的性质。
因为我们需要的树是静态的,所以构建笛卡尔树的复杂度可以降到O(n),具体构建的方法和虚树基本一致,就用栈维护一下当前的右链就可以了。
每次插入节点时一直弹栈,知道弹不动了,就向左连边,否则就连右边。
然后dp就变得十分容易,设dp[i][j]表示i子树内放j个的方案数,注意:我们每往下递归一层,宽度就要减去h[i],这样方便转移。
答案可能来自三部分,左子树、右子树和自己,转移时讨论一下就好了。
代码
#include<iostream> #include<cstdio> #define N 502 using namespace std; const int mod=1e9+7; typedef long long ll; const int maxn=1000000; ll dp[N][N],jie[maxn+2],ni[maxn+2]; int ch[N][2],a[N],n,size[N],k,st[N],top; inline int rd(){ int x=0;char c=getchar();bool f=0; while(!isdigit(c)){if(c=='-')f=1;c=getchar();} while(isdigit(c)){x=(x<<1)+(x<<3)+(c^48);c=getchar();} return f?-x:x; } inline ll power(ll x,ll y){ ll ans=1; while(y){if(y&1)ans=ans*x%mod;x=x*x%mod;y>>=1;} return ans; } inline ll C(int n,int m){if(n<m)return 0;return jie[n]*ni[m]%mod*ni[n-m]%mod;} void dfs(int u,int fa){ size[u]=1; if(ch[u][0])dfs(ch[u][0],u),size[u]+=size[ch[u][0]];if(ch[u][1])dfs(ch[u][1],u),size[u]+=size[ch[u][1]]; for(int i=0;i<=size[u];++i) for(int j=0;j<=i;++j)(dp[u][i]+=dp[ch[u][0]][j]*dp[ch[u][1]][i-j]%mod)%=mod; for(int i=size[u];i>=0;--i) for(int j=0;j<i;++j)(dp[u][i]+=dp[u][j]*jie[i-j]%mod*C(size[u]-j,i-j)%mod*C(a[u]-a[fa],i-j)%mod)%=mod; } int main(){ n=rd();k=rd(); dp[0][0]=1; for(int i=1;i<=n;++i)a[i]=rd(); jie[0]=1; for(int i=1;i<=maxn;++i)jie[i]=jie[i-1]*i%mod;ni[maxn]=power(jie[maxn],mod-2); for(int i=maxn-1;i>=0;--i)ni[i]=ni[i+1]*(i+1)%mod; for(int i=1;i<=n;++i){ while(top&&a[i]<a[st[top]]){ int x=st[top];top--; if(top&&a[i]<a[st[top]])ch[st[top]][1]=x; else ch[i][0]=x; } st[++top]=i; } while(top>1){ch[st[top-1]][1]=st[top];top--;} dfs(st[1],0); printf("%lld",dp[st[1]][k]); return 0; }