[Luogu P2523] [BZOJ 2302] [HAOI2011]Problem c

洛谷传送门

BZOJ传送门

题目描述

给$n$个人安排座位，先给每个人一个$1\sim n$的编号，设第$i$个人的编号为$a_i$（不同人的编号可以相同），接着从第一个人开始，大家依次入座，第$i$个人来了以后尝试坐到$a_i$，如果$a_i$被占据了，就尝试$a_{i+1}$，$a_{i+1}$也被占据了的话就尝试$a_{i+2}$，……，如果一直尝试到第$n$个都不行，该安排方案就不合法。然而有$m$个人的编号已经确定(他们或许贿赂了你的上司…)，你只能安排剩下的人的编号，求有多少种合法的安排方案。由于答案可能很大，只需输出其除以$M$后的余数即可。

输入输出格式

输入格式：

第一行一个整数$T$，表示数据组数

对于每组数据，第一行有三个整数，分别表示$n$、$m$、$M$

若$m$不为$0$，则接下来一行有$m$对整数，$p_1$、$q_1$，$p_2$、$q_2$ ,…, $p_m$、$q_m$，其中第$i$对整数$p_i$、$q_i$表示第$p_i$个人的编号必须为$q_i$

输出格式：

对于每组数据输出一行，若是有解则输出YES，后跟一个整数表示方案数$modM$，注意，$YES$和数之间只有一个空格，否则输出$NO$

输入输出样例

输入样例#1：

2
4 3 10
1 2 2 1 3 1
10 3 8882
7 9 2 9 5 10

输出样例#1：

YES 4
NO

说明

100%的数据满足：$1\le T\le 10，1\le n\le 300，0\le m\le n，2\le M\le 10^9，1\le p_i、q_i\le n$ 且保证$p_i$互不相同。

解题分析

先考虑如何判无解。 显然我们做一个后缀和数组$lim[i]$， 表示编号大于等于$i$的人的个数。 如果有一个$lim[i]$的值大于等于$n-i+1$， 那么显然后面是没有分配方案的， 直接输出$NO$。

有了这个限制， 我们就可以从后向前$dp$了。 设$dp[i][j]$表示考虑到第$i$个位置， 已经放下去$j$个， 那么直接枚举这个位置放多少个即可。

代码如下：

#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <cstdlib>
#include <cctype>
#define R register
#define IN inline
#define W while
#define gc getchar()
#define MX 305
#define ll long long
template <class T>
IN void in(T &x)
{
	x = 0; R char c = gc;
	for (; !isdigit(c); c = gc);
	for (;  isdigit(c); c = gc)
	x = (x << 1) + (x << 3) + c - 48;
}
int n, m, MOD, T, bd;
ll C[MX][MX], dp[MX][MX];
int lim[MX];
int main(void)
{
	int foo, bar, arr; in(T);
	st: W (T--)
	{
		std::memset(lim, 0, sizeof(lim));
		in(n), in(m), in(MOD); bd = n - m;
		for (R int i = 0; i <= 300; ++i) C[i][0] = 1;
		for (R int i = 1; i <= 300; ++i)
		for (R int j = 1; j <= i; ++j)
		C[i][j] = (C[i - 1][j - 1] + C[i - 1][j]) % MOD;
		for (R int i = 1; i <= m; ++i)
		in(foo), in(bar), lim[bar]++;
		for (R int i = n; i; --i)
		{
			lim[i] += lim[i + 1];
			if (lim[i] > n - i + 1)
			{puts("NO"); goto st;}
		}
		std::memset(dp, 0, sizeof(dp));
		dp[n + 1][0] = 1;
		for (R int i = n; i; --i)
		{
			arr = n - i + 1 - lim[i];
			for (R int j = 0; j <= arr; ++j)
			{
				for (R int k = 0; k <= j; ++k)
				(dp[i][j] += dp[i + 1][k] * C[bd - k][j - k] % MOD) %= MOD;
			}
		}
		printf("YES %lld\n", dp[1][bd]);
	}
}