Address
Meaning
称一个
1
∼
n
1\sim n
1∼n 的排列
P
P
P 的完美数为:有多少个
i
i
i 满足
∣
P
i
−
i
∣
=
1
|P_i-i|=1
∣Pi−i∣=1 。
求有多少个长度为
n
n
n 的完美数恰好为
m
m
m 的排列。
Solution
考虑容斥来做。具体地,设
s
(
k
)
s(k)
s(k) 表示在
[
1
,
n
]
[1,n]
[1,n] 中选出
k
k
k 个数,让它们中的每一个数
i
i
i 都满足
∣
P
i
−
i
∣
=
1
|P_i-i|=1
∣Pi−i∣=1 ,剩下的
n
−
k
n-k
n−k 个数随便排的方案数。
那么答案为:
∑
i
=
m
n
(
−
1
)
i
−
m
C
i
m
s
(
i
)
\sum_{i=m}^n(-1)^{i-m}C_i^ms(i)
i=m∑n(−1)i−mCims(i)
考虑通过 dp 求
s
s
s 。
f
[
i
]
[
j
]
[
0
/
1
]
[
0
/
1
]
[
0
/
1
]
f[i][j][0/1][0/1][0/1]
f[i][j][0/1][0/1][0/1] 表示在
[
1
,
i
]
[1,i]
[1,i] 内选出
j
j
j 个数放进
P
P
P 里面(满足与自己的位置差的绝对值为
1
1
1 ),最后三维表示
P
i
−
1
P_{i-1}
Pi−1 ,
P
i
P_i
Pi ,
P
i
+
1
P_{i+1}
Pi+1 三个位置是否已经有数填入。特别地,如果位置不存在则这一维只能为
0
0
0 。
边界
f
[
0
]
[
0
]
[
0
]
[
0
]
[
0
]
=
1
f[0][0][0][0][0]=1
f[0][0][0][0][0]=1 。
转移(1):
i
i
i 不被放进
P
P
P 内(作为剩下来的数随便排)。
f
[
i
+
1
]
[
j
]
[
o
p
2
]
[
o
p
3
]
[
0
]
+
=
f
[
i
]
[
j
]
[
o
p
1
]
[
o
p
2
]
[
o
p
3
]
f[i+1][j][op_2][op_3][0]+=f[i][j][op_1][op_2][op_3]
f[i+1][j][op2][op3][0]+=f[i][j][op1][op2][op3]
转移(2):如果
i
>
0
i>0
i>0 且
i
i
i 位置还没有数则可以把
i
+
1
i+1
i+1 放在
P
i
P_i
Pi 。
f
[
i
+
1
]
[
j
+
1
]
[
1
]
[
o
p
3
]
[
0
]
+
=
f
[
i
]
[
j
]
[
o
p
1
]
[
0
]
[
o
p
3
]
f[i+1][j+1][1][op_3][0]+=f[i][j][op_1][0][op_3]
f[i+1][j+1][1][op3][0]+=f[i][j][op1][0][op3]
转移(3):如果
i
<
n
−
1
i<n-1
i<n−1 且
i
+
2
i+2
i+2 位置还没有数则可以把
i
+
1
i+1
i+1 放在
P
i
+
2
P_{i+2}
Pi+2 。
f
[
i
+
1
]
[
j
+
1
]
[
o
p
2
]
[
o
p
3
]
[
1
]
+
=
f
[
i
]
[
j
]
[
o
p
1
]
[
o
p
2
]
[
o
p
3
]
f[i+1][j+1][op_2][op_3][1]+=f[i][j][op_1][op_2][op_3]
f[i+1][j+1][op2][op3][1]+=f[i][j][op1][op2][op3]
如果在
n
n
n 个数中选出
k
k
k 个数使得其中每个数
i
i
i 都满足
∣
P
i
−
i
∣
=
1
|P_i-i|=1
∣Pi−i∣=1 ,则在计算
s
(
k
)
s(k)
s(k) 时剩下
n
−
k
n-k
n−k 个随便排。
s
(
k
)
=
(
n
−
k
)
!
∑
o
p
1
∈
{
0
,
1
}
∑
o
p
2
∈
{
0
,
1
}
f
[
n
]
[
k
]
[
o
p
1
]
[
o
p
2
]
[
0
]
s(k)=(n-k)!\sum_{op_1\in\{0,1\}}\sum_{op_2\in\{0,1\}}f[n][k][op_1][op_2][0]
s(k)=(n−k)!op1∈{0,1}∑op2∈{0,1}∑f[n][k][op1][op2][0]
复杂度
O
(
n
2
)
O(n^2)
O(n2) 。
Code
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define For(i, a, b) for (i = a; i <= b; i++)
using namespace std;
const int N = 1005, ZZQ = 1e9 + 7;
int n, m, fac[N], C[N][N], f[N][N][2][2][2], ans;
int main()
{
int i, j, op1, op2, op3;
cin >> n >> m;
fac[0] = 1;
For (i, 1, n) fac[i] = 1ll * fac[i - 1] * i % ZZQ;
For (i, 0, n) C[i][0] = 1;
For (i, 1, n) For (j, 1, i)
C[i][j] = (C[i - 1][j] + C[i - 1][j - 1]) % ZZQ;
f[0][0][0][0][0] = 1;
For (i, 0, n - 1) For (j, 0, i)
{
For (op1, 0, 1) For (op2, 0, 1) For (op3, 0, 1)
(f[i + 1][j][op2][op3][0] += f[i][j][op1][op2][op3]) %= ZZQ;
if (i > 0) For (op1, 0, 1) For (op3, 0, 1)
(f[i + 1][j + 1][1][op3][0] += f[i][j][op1][0][op3]) %= ZZQ;
if (i < n - 1) For (op1, 0, 1) For (op2, 0, 1) For (op3, 0, 1)
(f[i + 1][j + 1][op2][op3][1] += f[i][j][op1][op2][op3]) %= ZZQ;
}
For (i, m, n)
{
int tmp = 0;
For (op1, 0, 1) For (op2, 0, 1)
tmp = (tmp + f[n][i][op1][op2][0]) % ZZQ;
int delta = 1ll * tmp * fac[n - i] % ZZQ * C[i][m] % ZZQ;
if (i - m & 1) ans = (ans - delta + ZZQ) % ZZQ;
else ans = (ans + delta) % ZZQ;
}
cout << ans << endl;
return 0;
}