题目链接:https://www.luogu.com.cn/problem/AT2366
有一个长度为 2 n − 1 2n-1 2n−1的序列 a a a,你可以将其重新排列,定义 b i b_i bi为 a 1 ∼ 2 i − 1 a_{1\sim 2i-1} a1∼2i−1的中位数。
询问有多少种不同的可能的 b b b序列。
1 ≤ n ≤ 50 1\leq n\leq 50 1≤n≤50
先假设 a a a中所有数字都不同,我们去考虑一下 b b b的限制。
相当于我们每次在 a i a_i ai中加入两个数字 x , y x,y x,y,然后如果 x , y x,y x,y都小于最后一个 b b b则新的 b b b取它在 a a a中的前驱,如果都大于则取后继,如果一大一小则不动。
基于这个限制我们得到的条件是每次 b i b_i bi加入数字之后的排名只能变动一位,这是充要的。
考虑去形式化这个条件,将 a a a排序后,我们有:
这样我们其实并不需要考虑 a a a填的情况,如果 b b b一次向外跨过了很多个 a a a,那么我们可以视这些 a a a此时还没有加入,我们只需要限制往内跨的情况出现就好了。
不过这第二个条件依旧不好处理,我们考虑倒着做,那当我们确定一个 b i b_i bi和 b i + 1 b_{i+1} bi+1之后,相当于 ( b i , b i + 1 ) (b_i,b_{i+1}) (bi,bi+1)之间的数字就都不能选择了。
考虑 d p dp dp,设 f i , l , r f_{i,l,r} fi,l,r表示目前填了 i i i个,对于现在的 b i b_i bi来说,左边还剩下 l l l个位置,右边还剩下 r r r个位置。
然后每次加入两边的 a i a_i ai,如果 a l = a l + 1 a_{l}=a_{l+1} al=al+1,那么我们将其视为同一个,右边同理。
时间复杂度: O ( n 4 ) O(n^4) O(n4)
#include
#include
#include
#define ll long long
using namespace std;
const ll N=110,P=1e9+7;
ll n,m,a[N],f[N][N][N];
signed main()
{
scanf("%lld",&n);m=2*n-1;
for(ll i=1;i<=m;i++)
scanf("%lld",&a[i]);
sort(a+1,a+m+1);
f[n][0][0]=1;
for(ll i=n-1;i>=1;i--){
ll l=i,r=2*n-i;
ll dl=(a[l]!=a[l+1]),dr=(a[r]!=a[r-1]);
for(ll l=0;l<=m;l++)
for(ll r=0;r<=m;r++){
if(!f[i+1][l][r])continue;
(f[i][l+dl][r+dr]+=f[i+1][l][r])%=P;
for(ll k=0;k<l+dl;k++)(f[i][k][r+dr+1]+=f[i+1][l][r])%=P;
for(ll k=0;k<r+dr;k++)(f[i][l+dl+1][k]+=f[i+1][l][r])%=P;
}
}
ll ans=0;
for(ll i=0;i<=m;i++)
for(ll j=0;j<=m;j++)
(ans+=f[1][i][j])%=P;
printf("%lld\n",ans);
return 0;
}