洛谷传送门
BZOJ传送门
题目描述
给 n n n个人安排座位,先给每个人一个 1 ∼ n 1\sim n 1∼n的编号,设第 i i i个人的编号为 a i a_i ai(不同人的编号可以相同),接着从第一个人开始,大家依次入座,第 i i i个人来了以后尝试坐到 a i a_i ai,如果 a i a_i ai被占据了,就尝试 a i + 1 a_{i+1} ai+1, a i + 1 a_{i+1} ai+1也被占据了的话就尝试 a i + 2 a_{i+2} ai+2,……,如果一直尝试到第 n n n个都不行,该安排方案就不合法。然而有 m m m个人的编号已经确定(他们或许贿赂了你的上司…),你只能安排剩下的人的编号,求有多少种合法的安排方案。由于答案可能很大,只需输出其除以 M M M后的余数即可。
输入输出格式
输入格式:
第一行一个整数 T T T,表示数据组数
对于每组数据,第一行有三个整数,分别表示 n n n、 m m m、 M M M
若 m m m不为 0 0 0,则接下来一行有 m m m对整数, p 1 p_1 p1、 q 1 q_1 q1, p 2 p_2 p2、 q 2 q_2 q2 ,…, p m p_m pm、 q m q_m qm,其中第 i i i对整数 p i p_i pi、 q i q_i qi表示第 p i p_i pi个人的编号必须为 q i q_i qi
输出格式:
对于每组数据输出一行,若是有解则输出YES,后跟一个整数表示方案数 m o d M mod\ M mod M,注意, Y E S YES YES和数之间只有一个空格,否则输出 N O NO 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 ≤ T ≤ 10 , 1 ≤ n ≤ 300 , 0 ≤ m ≤ n , 2 ≤ M ≤ 1 0 9 , 1 ≤ p i 、 q i ≤ n 1≤T≤10,1≤n≤300,0≤m≤n,2≤M≤10^9,1≤p_i、q_i≤n 1≤T≤10,1≤n≤300,0≤m≤n,2≤M≤109,1≤pi、qi≤n 且保证 p i p_i pi互不相同。
解题分析
先考虑如何判无解。 显然我们做一个后缀和数组 l i m [ i ] lim[i] lim[i], 表示编号大于等于 i i i的人的个数。 如果有一个 l i m [ i ] lim[i] lim[i]的值大于等于 n − i + 1 n - i+1 n−i+1, 那么显然后面是没有分配方案的, 直接输出 N O NO NO。
有了这个限制, 我们就可以从后向前 d p dp dp了。 设 d p [ i ] [ j ] dp[i][j] dp[i][j]表示考虑到第 i i i个位置, 已经放下去 j j 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]);
}
}