[Codeforces 285E]Positions in Permutations(容斥+DP)

本文详细解析洛谷RemoteJudge上一道经典题目——Codeforces285E,探讨如何求解长度为n的排列中,完美数恰好为m的排列数量。采用容斥原理和动态规划方法,实现O(n^2)复杂度的算法。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

Address

洛谷RemoteJudge
Codeforces 285E

Meaning

称一个 1 ∼ n 1\sim n 1n 的排列 P P P 的完美数为:有多少个 i i i 满足 ∣ P i − i ∣ = 1 |P_i-i|=1 Pii=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 Pii=1 ,剩下的 n − k n-k nk 个数随便排的方案数。
那么答案为:
∑ i = m n ( − 1 ) i − m C i m s ( i ) \sum_{i=m}^n(-1)^{i-m}C_i^ms(i) i=mn(1)imCims(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} Pi1 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 &lt; n − 1 i&lt;n-1 i<n1 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 Pii=1 ,则在计算 s ( k ) s(k) s(k) 时剩下 n − k n-k nk 个随便排。
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)=(nk)!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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值