专栏文章

延迟决策技巧

算法·理论参与者 13已保存评论 14

文章操作

快速查看文章及其快照的属性,并进行相关操作。

当前评论
14 条
当前快照
1 份
快照标识符
@mhz4qygq
此快照首次捕获于
2025/11/15 01:26
3 个月前
此快照最后确认于
2025/12/02 00:36
3 个月前
查看原文
  • 给每个对象决策属性,贡献与每个对象的属性相关。
  • 如果直接 DP 决策每个对象具体的属性,难以找到一个顺序,使得前面决策的属性不会再影响贡献。
  • 有一个单调的简单标准,把所有决策分作两部分:一部分是确定不会再影响贡献的(确定部分)、一部分是有可能会再影响贡献的(待定部分)。已经进入确定部分的决策不会再回到待定部分。
  • 确定部分等价。待定部分等价。
  • 此时可以考虑记录这个标准,下一个对象的决策分为 "让它的属性作为确定部分" 或 "让它的属性作为待定部分"。同时在标准发生改变的时候再处理那些从待定部分进入确定部分的决策。
一个经典的贡献就是 mex,按什么顺序决策都有可能出现先 1100 导致 mex 直接变到 22。但是 mex 随着出现的数的集合扩大,显然是单调不降的。而且若决策一个数 <mex<\text{mex} 是不会影响 mex 的。

例一:P7213
先考虑按什么顺序。显然要么按值域来看,要么按顺序来看。但是如果按值域来看需要同时处理两个位置,所以我们考虑按顺序。那是从前往后还是从后往前呢?注意前面的数是否会减少,要依赖后面的数,所以我们应该从后往前、这样记录的信息就可以判定当前数要怎么减少。
然后来考察性质。首先如果后面出现了一个数 xx,那么前面的数不可能等于 xx。进一步观察,我们发现操作其实可以不看作是每一轮都对整体操作,而是转为:从后往前扫,只要当前的数后面出现过就 1-1,直到变成 00 或者后面没有。扫完了就得到最终的结果。
所以一个数 xx 要变成 00,等价于后面 1x1\sim x 都出现了。那么我们可以大致得出一个数如果填 xx 就有两种可能:
  • 若后面 1x1\sim x 都出现了,变成 00
  • 若后面 1x1\sim x 里有漏的,变成漏的里最大的。
那么难点就在于我们怎么判断 1x1\sim x 里有没有漏的。如果具体决策每个数等于几,那就难以避免状态每个数是否出现。但是此时发现有一个单调性:如果已经连续出现了 1x1\sim x,那么未来连续出现的上限只会比 xx 大。所以决策一个数为 x\le x 不会改变这个连续高度,而决策 >x>x 有可能改变
把当前后缀里的数分为两个部分:1h1\sim hh+2\ge h+2 的数。考虑在 DP 过程中记录连续出现的高度,决策下一个数是 h\le h 还是 h+2\ge h+2,前者立刻算贡献,后者待定。当 hh 发生变化时,决策待定的元素里有哪些要被确定。
dp(i,h)dp(i,h) 表示 ini\sim n 的高度决策好了,最大连续出现高度为 1h1\sim h 的方案数。
此时我们应该要预见到(或者在下面推的时候碰到):一种高度只有两个名额,当剩下两个名额或者一个名额,可以做 11 的贡献;如果不剩下名额,就做 00 的贡献。而从两个名额变成一个名额、和从一个名额变成没有名额,这两种变化是本质不同的!所以难道我们要额外记录有多少个高度是两个名额的吗?
这样做复杂度肯定爆炸了。考虑上面遇到的问题,我们可以认为相等的两个数也是有区分的,最终给答案除以 2n2^n。这样有两个名额的贡献系数就是 22,从而转为有多少个名额可选,贡献系数就是多少。
然后考虑转移到 i1i-1。设 ini\sim n 里有 pip_i 个要变成 00qi=ni+1piq_i=n-i+1-p_i 个要留下。
  • i1i-1 要变成 00,则 i1i-1 的高度要 h\le h
    一共有 2h2h 个名额,但是有 hh 个作为 1h1\sim h 的连续前缀,以及 pip_i 个变成 00 的初始高度肯定 h\le h,因此被占用了 h+pih+p_i 个名额,剩下 hpih-p_i 个。
    所以 dp(i,h)×(hpi)dp(i1,h)dp(i,h)\times (h-p_i)\rightarrow dp(i-1,h)
  • i1i-1 要留下。
    • i1i-1 最终高度 >h+1>h+1
      那不会更新 hh,先放着在以后考虑。dp(i,h)dp(i1,h)dp(i,h)\rightarrow dp(i-1,h)
    • i1i-1 最终高度 =h+1=h+1
      考虑枚举加入 i1i-1 后新的连续出现高度 hh'。那么要在后面的自由高度里选 h(h+1)h'-(h+1) 个来填补空缺。
      所以后面的数又分成两部分:h\le h' 的、>h+1>h'+1 的。
      考虑贡献系数。
      1. 先选出哪些位置要来补空缺,方案数 (qihh(h+1))\binom{q_i-h}{h'-(h+1)}
      2. 然后给 i1i-1 决策一个初始高度。
        一共 2(hh)2(h'-h) 个名额,后面要用 h(h+1)h'-(h+1) 个名额,还有 hh+1h'-h+1 个名额。
        方案数 hh+1h'-h+1
      3. 最后还要考虑后面的位置怎么选使得能刚好消到 h+2hh+2\sim h'
        显然这个问题等价于 1,1,2,2,,hh1,hh11,1,2,2,\dots,h'-h-1,h'-h-1 里选 hh1h'-h-1 个数出来,操作完了刚好是 1hh11\sim h'-h-1 的排列的方案数。
        但是这个东西不是很好直接求,所以设辅助数组 gng_n 表示 "在 1,1,2,2,,n,n1,1,2,2,\dots,n,n 里选 nn 个数,操作完了剩下 1n1\sim n 的排列的方案数"。
        考虑怎么求 gg。如何划分状态?
        我们考虑极端值原理,枚举最终 11 号位的高度 xx,那么其原本高度在 [x,n][x,n] 里有 2(nx+1)2(n-x+1) 个名额,而之前用了 nxn-x 个在 [x+1,n][x+1,n] 里,所以还剩 nx+2n-x+2 个名额。 同时我们发现后面最终高度为 1x11\sim x-1x+1nx+1\sim n 的不会相互干扰了(因为没有 xx 供给它们跨越中间),所以有转移:
        gn=x(nx+2)(n1x1)gx1gnxg_n=\sum_x(n-x+2)\binom{n-1}{x-1}g_{x-1}g_{n-x}
        方案数 ghh1g_{h'-h-1}
      所以 dp(i,h)×(qihh(h+1))(hh+1)ghh1dp(i1,h)dp(i,h)\times \binom{q_i-h}{h'-(h+1)}(h'-h+1)g_{h'-h-1}\rightarrow dp(i-1,h')
转移即可,复杂度 O(n3)O(n^3)。不要忘记最终除以 2n2^n
在本题中,单调的标准即 "连续出现高度 hh"。如果决策 h\le h,不会改变 hh;而决策 h+2\ge h+2 可能在后面改变 hh,也可能不会影响。于是待定后者的决策即可。

例二:P14364
把计数排列看作天和人的匹配。按值域考虑每个人和哪个匹配。大致为 dp(i,j)dp(i,j) 表示 cic\le i 的人里有 jj 个面试失败的方案数。
然后就发现当决策一个人面试失败的时候并不只涉及 cc 比它小的人,因为可以让一个 cc 大的人遇到 00,所以还要额外记录后面有多少个人面试失败了。
为了避免记录每个人的状态,我们要考虑这个单调的标准,从而把同属于 "确定" 或 "待定" 的人能同类看待。
(i,cpi)(i,c_{p_i}) 在坐标系上画成柱状图,画一条左下到右上的折线表示每一天的失败人数变化。这样就容易看出因为折线是单调向上的,所以在某次折线达到 xx 的高度后,之后耐心值 x\le x 的人都不可能被录取,而 >x>x 的人只要放在 11 的位置就一定会被录取。这就是单调的标准
所以考虑把人分作耐心值 x,>x\le x,>x 考虑。因为 xx 是单调上升的,所以过程中会有一些 >x>x 的人进入 x\le x,但是 x\le x 的永远会 x\le x。所以 "x\le x 的人" 是确定的部分,">x>x 的人" 是不确定的部分。考虑设 dp(i,j,k)dp(i,j,k) 表示决策前 ii 天的安排、有 jj 个人失败、有 kk 个位置的人耐心值 >j>j(等待后面决策),在 dpdp 里只决策前 ii 天耐心值 j\le j 的方案数。
使用刷表法转移。考虑第 i+1i+1 天的人是要失败还是要通过。记 cnticnt_i 表示耐心值为 ii 的人数,scsccntcnt 的前缀和。
  • si+1=0s_{i+1}=0
    必须失败。分类讨论 cpi+1c_{p_{i+1}}j+1j+1 的大小关系,并把 kk 个人里 =j+1=j+1 的纳入考虑。
    • cpi+1>j+1c_{p_{i+1}}>j+1
      w=0k(kw)(cntj+1w)w!dp(i,j,k)dp(i+1,j+1,kw+1)\sum_{w=0}^k\binom{k}{w}\binom{cnt_{j+1}}{w}w!dp(i,j,k)\rightarrow dp(i+1,j+1,k-w+1)
    • cpi+1j+1c_{p_{i+1}}\le j+1
      w=0k(kw)(cntj+1w)w!(scj+1(i(kw))dp(i,j,k)dp(i+1,j+1,kw)\sum_{w=0}^k\binom{k}{w}\binom{cnt_{j+1}}{w}w!(sc_{j+1}-(i-(k-w))dp(i,j,k)\rightarrow dp(i+1,j+1,k-w)
  • si+1=1s_{i+1}=1
    • cpi+1>jc_{p_{i+1}}>j,此时会录取,jj 不变。
      dp(i,j,k)dp(i+1,j,k+1)dp(i,j,k)\rightarrow dp(i+1,j,k+1)
    • cpi+1jc_{p_{i+1}}\le j,此时会放弃,jj 加一。
      w=0k(kw)(cntj+1w)w!(scj+1(ik))dp(i,j,k)dp(i+1,j+1,kw)\sum_{w=0}^k\binom{k}{w}\binom{cnt_{j+1}}{w}w!(sc_{j+1}-(i-k))dp(i,j,k)\rightarrow dp(i+1,j+1,k-w)
虽然看起来要枚举 wwO(n4)O(n^4) 的,但因为总人数 O(n)O(n),所以 ww 一共只能枚举 nn 次。故总复杂度是 O(n3)O(n^3) 的。

例三:CF1608F
考虑转换计数对象,令 sis_iaa 的前缀 mex\text{mex} 数组,对 ss 计数再考察一个 ss 可以对应多少个 aa
对于 ss,显然应该有 siis_i\le isisi+1s_i\le s_{i+1}si[bik,bi+k]s_i\in [b_i-k,b_i+k]
考虑固定 ssaa 有多少种选取方案。手玩一下,首先能发现一个简单的性质:sisi+1s_i\neq s_{i+1} 时,ai+1=sia_{i+1}=s_i。于是这启示我们把 ss 分成若干个相等段来考虑。
于是设划分的结果是 [l1=1,r1],[l2=r1+1,r2],,[lk,rk=n][l_1=1,r_1],[l_2=r_1+1,r_2],\dots,[l_k,r_k=n]。每段内的 ss 都相等。那么有:
  • 2ik2\le i\le kali=sri1a_{l_i}=s_{r_{i-1}}
  • 1jri1\le j\le r_iajsria_j\neq s_{r_i}
  • 为了使 alia_{l_i} 一加入,就从 sri1s_{r_{i-1}} 增加到 slis_{l_i},则 a1ari1a_1\sim a_{r_{i-1}} 必须包含 (sri1,sli)(s_{r_{i-1}},s_{l_i}) 的所有数。
  • 如果 s1=1s_1=1a1=0a_1=0
感受一下,以上的条件已经充分了。所以开始 DP:因为有 mex\text{mex} 运算,所以采用延迟决策、记录待定位置数的方法,f(i,j,k)f(i,j,k) 表示前 ii 个位置,si=bi+js_i=b_i+j,有 kk 个位置上的 aa 是没确定的(>si>s_i 的个数),方案数。
考虑转移。枚举下一位 si+1=bi+1+ps_{i+1}=b_{i+1}+pp[k,k]p\in [-k,k]):
  • si=si+1s_i=s_{i+1},则 f(i,j,k)f(i+1,p,k+1)f(i,j,k)\rightarrow f(i+1,p,k+1)
  • si<si+1s_i<s_{i+1}。有 ai+1=sia_{i+1}=s_i,还要从前面待定的 kk 个位置里选 kk' 个,赋值到 (si,si+1)(s_i,s_{i+1}) 里,且每个数都要取到,这里面是一个要推的系数。
就算假设那个系数可以 O(1)O(1) 求,状态是 O(n2k)O(n^2k) 的,枚举 si+1s_{i+1}O(k)O(k) 的,若 si<si+1s_i<s_{i+1} 还要枚举放多少个下来还是 O(k)O(k) 的。复杂度是 O(n2k3)O(n^2k^3) 起步,肯定过不了,必须优化。
接下来是一个很有道理但是很巧妙的想法:mex\text{mex} 只关心每种数是否出现过,那如果我们待定的方式是有 kk 种数是 >si>s_i 的,就非常方便了。所以我们改设 f(i,j,k)f(i,j,k) 表示前 iisi=js_i=jkk 种数待定。
那么 si=si+1s_i=s_{i+1} 时,ai+1a_{i+1} 可以选择新开一种或者选择等于旧的一种。而 si<si+1s_i<s_{i+1} 时只需要从 kk 种里选 si+1si1s_{i+1}-s_i-1 种就行了!
所以更改了一下定义,轻松就做到了 O(n2k2)O(n^2k^2)。但是这个复杂度还是过不了。继续优化。写出转移式子:
f(i,j,k)f(i+1,j,k+1)f(i,j,k)\rightarrow f(i+1,j,k+1) f(i,j,k)(j+k)f(i+1,j,k)f(i,j,k)\cdot (j+k)\rightarrow f(i+1,j,k) f(i,j,k)(k(bi+1+j)(bi+j)1)f(i+1,j,k[(bi+1+j)(bi+j)1])f(i,j,k)\cdot \binom{k}{(b_{i+1}+j')-(b_i+j)-1}\\\rightarrow f(i+1,j',k-[(b_{i+1}+j')-(b_i+j)-1])
现在来优化。前两条不用优化,看看第三条是什么东西,发现枚举 jj' 完全不如直接枚举 bi+1+j(bi+j)1b_{i+1}+j'-(b_i+j)-1。重写一下就是 f(i,j,k)f(i,j,k) 会转移到 f(i+1,j+c,kc)f(i+1,j+c,k-c) 这条斜线上。用个前缀和优化即可。
upd:有一个聪明的方法可以不用前缀和优化。我们不用 "一次性" 决定好 mex\text{mex} 要增长到哪里。可以认为 mex\text{mex} 有两种增长方式:在 aia_i 放一个 mex\text{mex},或者在目前待定里取一种放 mex\text{mex}。所以可以写成 f(i,j,k)f(i1,j1,k)+f(i,j1,k+1)×(k+1)f(i,j,k)\leftarrow f(i-1,j-1,k)+f(i,j-1,k+1)\times (k+1)

例四:图灵杯第七届高级组T1 棋无常树
mxbimxb_iii 子树内 bib_i 的最大值,则赋值后 ii 子树内 0mxbi10\sim mxb_i-1 都应该出现过。同时 ii 子树内 a1a\neq -1 的数也确定出现过。定义 visi,j=0/1vis_{i,j}=0/1 表示 ii 子树内数 jj 是否出现过。
定义 fi,j,k=0/1f_{i,j,k=0/1} 表示在 ii 子树内填 aa,使得 "mxbi\neq mxb_i 且不确定是否出现过的数里" 有 jj 种不同的数,且 mxbimxb_i 没有出现/出现了 的方案数。所以这个状态实际上把数分成了三类:确定的数、不确定的数、mxbxmxb_x在转移时就要让这三类数指代的对象统一了才能转移
考虑合并 xx 和其子树 ii,为了方便起见,不妨 mxbxmxbimxb_x\ge mxb_i。记 szusz_u 为当前 uu 子树内有多少个 a=1a=-1 的,也就是 ff 数组的第二维在有意义的前提下最高取到多少
因为 ff 的定义是基于 visvis 的,第一个任务是合并 fx,fif_x,f_ivisvis
当我们发现 visx,p=1,visi,p=0vis_{x,p}=1,vis_{i,p}=0 时,我们要把 visi,pvis_{i,p} 改成 11,那这会对 fif_i 有什么影响?pp 本来在 ii 的视角下属于 "不确定是否出现过的数",现在变成确定出现的了。所以:
fi,j,kfi,j,k+fi,j+1,kjf_{i,j,k}\leftarrow f_{i,j,k}+f_{i,j+1,k}\cdot j
也就是从 j+1j+1 个不确定的里面选一个 pp 出来变成确定的,这样更新一次是 O(szi)O(sz_i) 的。visx,p=0,visi,p=1vis_{x,p}=0,vis_{i,p}=1 的同理对 fxf_x 更新,一次 O(szx)O(sz_x)
这个操作会经常出现,我们称它为 trans。可以发现,当有一个新的确定数出现,为了确保不确定数和确定数之间没有重复,就需要做一次 trans。
然后 visx,visivis_x,vis_i 相等了,开始合并 fx,fif_x,f_i。为了方便,搞一个 gj,0/1g_{j,0/1} 作为临时数组,fxfif_x*f_i 先放到 gg 里面,转移完了再从 gg 复制回 fxf_x
  • 先考虑 mxbx=mxbimxb_x=mxb_i 的情况。一个 naive 的想法是 fx,j,afi,k,bgj+k,max(a,b)f_{x,j,a}\cdot f_{i,k,b}\rightarrow g_{j+k,\max(a,b)}。但这个转移方程是不对的,原因是 xx 里的 jj 种不同数可能与 ii 里的 kk 种不同数有重合,导致最终的不同种类数比 j+kj+k 少。
    枚举 j:0szxj:0\rightarrow sz_x。我们希望 fx,j,afi,k,bf_{x,j,a}\cdot f_{i,k,b} 可以直接转移,问题在于两棵子树内的不同数会有重合。所以一个想法是随着 jj 的变动,始终保持 fi,kf_{i,k}kk 个不同数里与 fx,jf_{x,j}jj 个数不重合
    那么当 jj 增大一,我们只需要处理这 kk 个不同数里出现了 xx 的第 jj 种数的情况。怎么处理?这和上面计算 "把 visi,pvis_{i,p} 改成 11,那这会对 fif_i 有什么影响" 是完全一样的。于是只需要 trans 一遍即可。
  • 那么只剩下 mxbx>mxbimxb_x>mxb_i(开头的时候有不妨 mxbxmxbimxb_x\ge mxb_i)。
    容易发现上面 jj 增大一就对 fif_i 做一次 trans 的想法是可以沿用的。但是现在又增加了一个新的问题:fi,k,0/1f_{i,k,0/1} 的第三维 0/10/1 指的是 mxbimxb_i 而非 mxbxmxb_x。要想让 fif_i 转移到 xx 的子树内,又要分类讨论 fif_i 里有没有 mxbxmxb_x
    • 如果 visi,mxbx=1vis_{i,mxb_x}=1,那么显然第三维只能取 11。也就是 fx,j,afi,k,bgj+k,1f_{x,j,a}\cdot f_{i,k,b}\rightarrow g_{j+k,1}
    • 否则,如果 fif_i 里没有 mxbimxb_i:转移方程 fx,j,afi,k,bgj+k,af_{x,j,a}\cdot f_{i,k,b}\rightarrow g_{j+k,a}
      如果 fif_i 里有 mxbimxb_i:转移方程 fx,j,a(kfi,k,b)gj+k1,1f_{x,j,a}\cdot (k\cdot f_{i,k,b})\rightarrow g_{j+k-1,1}
我们终于合并了所有子树,太好了!但是现在还需要把 xx 的影响加入 fxf_x 内。这又要根据 ax,bxa_x,b_x 分类讨论。令 ss 表示 xx 子树内有多少种确定的数会出现(s=ivisx,is=\sum_i vis_{x,i})。
  • ax=bx=1a_x=b_x=-1
    考虑 axa_x 赋值成什么:
    • 赋值成一个新的不确定数。fx,j+1,afx,j+1,a+fx,j,af_{x,j+1,a}\leftarrow f_{x,j+1,a}+f_{x,j,a}
    • 赋值成一个确定数或者一个旧的不确定数。fx,j,afx,j,a×(s+j)f_{x,j,a}\leftarrow f_{x,j,a}\times (s+j)
    • 赋值成 mxbxmxb_xfx,j,1fx,j,1+fx,j,kf_{x,j,1}\leftarrow f_{x,j,1}+f_{x,j,k}
    注意让 szxsz_x 加一。
  • ax1,bx=1a_x\neq -1,b_x=-1
    此时无法赋值成不确定数,只能是确定数或者 mxbxmxb_x
    • ax=mxbxa_x=mxb_x,那么 fxf_{x} 的第三维只能是 11(必然存在 mxbxmxb_x)。令 fx,j,1fx,j,1+fx,j,0f_{x,j,1}\leftarrow f_{x,j,1}+f_{x,j,0} 然后 fx,j,00f_{x,j,0}\leftarrow 0
    • 若在 xx 之前已经有 axa_x 了,那没有任何影响。
    • 否则 xx 是一个新出现的确定数,也就是 visx,ax1vis_{x,a_x}\leftarrow 1。为了排除 axa_x 与不确定数的重复,做一次 trans。
  • bx1b_x\neq -1
    mxbx>bxmxb_x>b_xax=bxa_x=b_xvisx,bx=1vis_{x,b_x}=1,那么无解,直接退出。
    • bx=mxbxb_x=mxb_x
      这种情况我们只接受 fx,?,0f_{x,?,0} 的转移。因为如果第三维是 11xx 的子树里就出现 bxb_x 了,无解。所以一开始就让所有 fx,j,10f_{x,j,1}\leftarrow 0
      • ax=1a_x=-1。可以赋值成确定数或旧的不确定数或新的不确定数。如果赋值成新的不确定数,fx,j+1,0fx,j+1,0+fx,j,0f_{x,j+1,0}\leftarrow f_{x,j+1,0}+f_{x,j,0};如果赋值成旧的不确定数或确定数,fx,j,0fx,j,0(s+j)f_{x,j,0}\leftarrow f_{x,j,0}\cdot (s+j)
        注意 szxsz_x 加一。
      • ax1a_x\neq -1。若 axa_x 以前出现过,不用管;否则 visx,ax1vis_{x,a_x}\leftarrow 1,然后为了排除 jj 个不确定数里可能有 axa_x 出现,再做一次 trans 即可。
    • bx>mxbxb_x>mxb_x
      这种情况下,最终的 mxbx=bxmxb_x=b_x,所以最终的 fxf_x 应该满足所有第三维是 11 的位置都是 00
      同时当 mxbx1mxb_x\neq -1,因为 bxb_x 要满足,[0,bx1][0,b_x-1] 都出现过,所以 mxbxmxb_x 也必须出现过。于是只能从原本第三维是 11fxf_x 转移。
      因此要特别注意,现在的 ff 第三维指的是 mxbxmxb_x,需要的 ff 第三维指向 bxb_x
      • mxbx=1mxb_x=-1。注意这种情况显然应该有 fxf_x 所有第三维是 11 的位置都是 00
        1. ax=1a_x=-1。可以赋值成新的不确定数或旧的不确定数或确定数。
          记得 szxsz_x 加一。
        2. ax1a_x\neq -1。若 axa_x 之前出现过,不用理;否则 visx,ax1vis_{x,a_x}\leftarrow 1 然后做一次 trans。
      • mxbx1mxb_x\neq -1
        1. ax=1a_x=-1
          如果赋值为一个新的不确定数,就是 fx,j+1,0fx,j+1,0+fx,j,1f_{x,j+1,0}\leftarrow f_{x,j+1,0}+f_{x,j,1}。注意这里只能从 fx,j,1f_{x,j,1} 出发转移。
          否则不确定数个数(第二维)不增加,也就是转成旧不确定数或确定数或 mxbxmxb_x。如果转成旧不确定数,贡献 jfx,j,1j\cdot f_{x,j,1};如果转成确定数,贡献 sfx,j,1s\cdot f_{x,j,1};如果转成 mxbxmxb_x,因为 axa_x 处有了 mxbxmxb_x,其他地方就可以使用 fx,j,0f_{x,j,0} 转移了,贡献 fx,j,0+fx,j,1f_{x,j,0}+f_{x,j,1}
          因此 fx,j,0fx,j,0+(s+j+1)fx,j,1f_{x,j,0}\leftarrow f_{x,j,0}+(s+j+1)\cdot f_{x,j,1}
          然后 fx,j,1=0f_{x,j,1}=0
          记得 sz[x]sz[x] 加一。
        2. ax1a_x\neq -1
          这里又要分为 ax=mxbxa_x=mxb_xaxmxbxa_x\neq mxb_x
          ax=mxbxa_x=mxb_x,那么其他地方有没有 mxbxmxb_x 都可以,fx,j,0fx,j,0+fx,j,1f_{x,j,0}\leftarrow f_{x,j,0}+f_{x,j,1},然后 fx,j,10f_{x,j,1}\leftarrow 0
          axmxbxa_x\neq mxb_x,那么其他地方必须有 mxbxmxb_x,所以 fx,j,0fx,j,1f_{x,j,0}\leftarrow f_{x,j,1},然后 fx,j,10f_{x,j,1}\leftarrow 0。同时注意如果 axa_x 是新的确定数出现,做一次 trans。
        最后,因为 bxb_x 要满足,还要让一些不确定数变成 [mxbx+1,bx1][mxb_x+1,b_x-1] 里的数。具体而言就是 fx,i1,jfx,i,jif_{x,i-1,j}\leftarrow f_{x,i,j}\cdot i,然后让 szxsz_x 减一。不要忘了让 mxbxbxmxb_x\leftarrow b_x
于是 DP 的部分结束了,怎么计算答案?
s=n+1ivis1,i(mxb11)s=n+1-\sum_i vis_{1,i}-(mxb_1\neq -1),那么 ss 就是不确定数的选择范围。
ansans+(si)i!f1,i,0/1ans\leftarrow ans + \binom{s}{i}i! \cdot f_{1,i,0/1}
然后是复杂度分析。
  • 合并 visvis 数组的复杂度。把 trans 的 O(szi)O(sz_i) 看作对子树内每个结点增加 11 的势能。对于结点 xx 不断向上合并,每增加 11 的势能就表示 visvis 出现了一个新的位置变成 11,而一共有 n+1n+1 个位置可以变成 11,所以一个点贡献的总复杂度是 O(n)O(n)。于是这一部分总共贡献 O(n2)O(n^2)
  • 合并 ff 数组的复杂度。把 O(szxszi)O(sz_x\cdot sz_i) 看作在两棵子树里各选择一个点,在 LCA 处贡献 11 的复杂度。容易分析出 O(n2)O(n^2)
  • 加入 xx 的复杂度。容易做到 O(szx)O(sz_x) 的复杂度。总共 O(n2)O(n^2)
因此总复杂度 O(n2)O(n^2)

评论

14 条评论,欢迎与作者交流。

正在加载评论...