https://atcoder.jp/contests/abc330
G - Inversion Squared
a[n]
$a[i] \in [1,n]$ 或$a[i]=-1$, 1~n最多出现一次
-1能替换成$[1\cdots n]$中的数
求 所有 ((a能产生的$1\dots n$排列)的逆序对的个数)的平方和,mod 998244353
n = 3000
2s
1024mb
我的思路
题面很质朴,n 3000说明可能n^2一类的算法
令x=-1的个数
对于非-1的部分可以直接算出,然后乘上 (x)! 就是贡献了
对于-1和-1之间,要知道每一个 排列的方案,交换这两个位置 对应的也是一个方案,而这两个方案之间 一个贡献1,一个贡献0
所以 是 $x!\frac{(x-1)x}{2}\cdot \frac{1}{2}$, 分别是 总方案数,配对位置数,和1/2的平均贡献
那么这个问题来到的就是 -1和给定值之间的逆序对 的贡献个数了
contribute[有值pos1] = f_left(value,left -1 count) + f_right(value, right -1 count)
所以 如果能 计算出
$f(value,cnt)=$ 在一共x个-1的情况下,左侧-1的位置有cnt个,当前值是value,的方案数
l[v] =
比v小的未填的值的个数
那么左侧的贡献为 $cnt\star (x-l[value]) (x-1)!$
右侧的贡献为 = $(x-cnt)\star lvalue!$
似乎就没了吧?
xxxxxxxxxxxxxxxxxx上面读错题目了, 要逆序对个数 的平方和,不只是逆序对的个数和
已经填好的逆序对=C,-1内部的逆序对=A,-1和填好的逆序对=B
$(C+A+B)^2=C^2+A^2+B^2+2AB+2BC+2CA$
首先C是不变的,A,B是根据填写情况变化的
所以这里的$C^2+2BC+2CA$很好算,和上面一样,
那问题来到$A^2,B^2,2AB$怎么算
一个想法是和上面一样,考虑两个位置交换会对贡献有什么影响
显然一样的, 交换2个位置,那么它们的A相差为1,问题是B是可能变化的,且变化无法确定
A^2 也有办法,因为A^2相当于x个数的全排列的 逆序对平方和
1 2 3 4
| len=1:0 len=2:0 [1] len=3:0 1 [1 2][2 3] len=4:0 1 1 2 2 3 [1 2 2 3 3 4][2 3 3 4 4 5][3 4 4 5 5 6]
|
个数上可以 生成函数表示
$f_0(x)=1$
$f_1(x)=1+x=(1+x)f_0(x)$
$f_2(x)=1+2x+2x^2+x^3=(1+x+x^2)f_1(x)$
$f_3(x)=(1+x+x^2+x^3)f_2(x)$
这个 采取 分治的方法计算 $O(n^2 \log n)$ 可以算出来,但是 $2^{23} =8388608 > 4498500=\frac{2999\cdot 3000}{2}$ 这会炸掉吗?
然而这个思路带来的一个新的启发能否用 生成函数的方法 直接计算出每个逆序对的方案数,
例如 $f_3(x)=(1+x+x^2+x^3)f_2(x)$ 中$x^2$是 在说 对于后缀长度$4$的4个数字,第一个是其中第2+1小的(因为1-index所以+1),
想了想,好像并不行,虽然在-1的位置填入值的时候,可以计算和后面相比多少个,但是这里产生的分叉状态没法合并,意味着分叉的状态数量会影响时间空间复杂度