题目
思路
说起来好像也不能叫做难题,但是我中途想错了两次!值得敲响警钟。
考虑从左往右一列一列的放棋子。想想 d p \tt dp dp 状态需要记录什么?即当前这一列上,有哪些格子是不能放的。这玩意儿显然很难计算,不状压根本做不了,因为 h h h 变小需要砍掉顶部的值。
有什么方法避免 “砍掉” 呢?显然是 h h h 从小到大处理。而且过程中一定要避免遇到 h h h 变小的情况。显然应该是建笛卡尔树,也就是按照 min h \min h minh 分治。
我的错误想法一:两边递归,然后选一些 放了的棋子 移动到最下面 h h h 行中。显然这样会重复。
我的错误想法二:将 “底部” 有多少行被占用了,也作为状态。结果这样两边不是独立子问题。
修正错误想法一:两边递归,然后选一些 没放棋子 的列,考虑它们放到最下面的哪几行。
显然合并两个子树的复杂度是 s i z e size size 的乘积,合并后需要 s i z e 2 size^2 size2 修改信息,所以总复杂度是 O ( n 3 ) \mathcal O(n^3) O(n3) 的,常数并不大。
代码
#include <cstdio>
#include <iostream>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
# define rep(i,a,b) for(int i=(a); i<=(b); ++i)
# define drep(i,a,b) for(int i=(a); i>=(b); --i)
typedef long long int_;
inline int readint(){
int a = 0; char c = getchar(), f = 1;
for(; c<'0'||c>'9'; c=getchar())
if(c == '-') f = -f;
for(; '0'<=c&&c<='9'; c=getchar())
a = (a<<3)+(a<<1)+(c^48);
return a*f;
}
inline void writeint(int_ x){
if(x > 9) writeint(x/10);
putchar((x-x/10*10)^48);
}
const int Mod = 1e9+7;
namespace Math{
const int MaxM = 1000005;
int jc[MaxM], inv[MaxM];
void import(int n = MaxM-1){
jc[0] = inv[0] = jc[1] = inv[1] = 1;
for(int i=2; i<=n; ++i){
jc[i] = 1ll*jc[i-1]*i%Mod;
inv[i] = (0ll+Mod-Mod/i)*inv[Mod%i]%Mod;
}
rep(i,2,n) inv[i] = 1ll*inv[i]*inv[i-1]%Mod;
}
inline int getC(int n,int m){
if(n < m || m < 0) return 0;
return 1ll*jc[n]*inv[m]%Mod*inv[n-m]%Mod;
}
}
using Math::getC;
const int MaxN = 505;
int h[MaxN], cntNode, n;
int dp[MaxN][MaxN], tmp[MaxN];
int solve(int l,int r,int d=0){
int o = ++ cntNode; // index
if(l == r){
dp[o][1] = h[l]-d;
dp[o][0] = 1; return o;
}
int mn = min_element(h+l,h+r+1)-h;
if(mn == l) o = solve(l+1,r,h[mn]);
if(mn == r) o = solve(l,r-1,h[mn]);
if(mn != l && mn != r){
int L = solve(l,mn-1,h[mn]);
int R = solve(mn+1,r,h[mn]);
rep(i,0,mn-l) rep(j,0,r-mn)
dp[o][i+j] = (dp[o][i+j]+
1ll*dp[L][i]*dp[R][j])%Mod;
}
d = h[mn]-d; // width of blank
memset(tmp+1,0,(r-l+1)<<2);
drep(i,min(d,r-l+1),0) // how many rows are added
rep(j,0,r-l+1-i) // count of columns used
tmp[j+i] = (tmp[j+i]+1ll*Math::jc[d]
*getC(r-l+1-j,i)%Mod*dp[o][j]
%Mod*Math::inv[d-i])%Mod;
memcpy(dp[o]+1,tmp+1,(r-l+1)<<2);
return o; // return index
}
int main(){
Math::import();
int n = readint(), k = readint();
rep(i,1,n) h[i] = readint();
int o = solve(1,n);
printf("%d\n",dp[o][k]);
return 0;
}