LOI 1386 座位安排

题目描述

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

分析

• 仔细想一下会发现其中给的$p_i$并没有什么用处，我们只需要知道那个位置之前有人占了，而不需要知道到底是谁占领了它。
• 然后我们考虑如何判断非合法的状态，记$s_{i}$表示在$i$位置后面被占领的位置，如果当$s_i>n-i+1$时显然是不满足题意的。
• 最后我们考虑转移的方程，记$f_{i,j}$表示元素值不小于$i$，有$j$个位置固定的方案数。
• 方程： $$f_{i,j} = \Sigma f_{i+1,j-k} * C_{j,k}$$
• 其中$j$的范围是$[0,N - s[i]-i +1]$
• 其中$k$的范围是$[0,j]$
• $C_{j,k}$表示从$j$个中选$k$个的组合数

#include <bits/stdc++.h>
#define rep( i , l , r ) for( int i = (l) ; i <= (r) ; ++i )
#define per( i , r , l ) for( int i = (r) ; i >= (l) ; --i )
#define erep( i , u ) for( int i = head[(u)] ; ~i ; i = e[i].nxt )
using namespace std;
int _read(){
    char ch = getchar();
    int x = 0 , f = 1 ;
    while( !isdigit( ch ) )
           if( ch == '-' ) f = -1 , ch = getchar();
           else ch = getchar();
    while( isdigit( ch ) )
           x = (ch  - '0') + x * 10 , ch =  getchar();
    return x * f;
}
typedef long long ll;
const int maxn = 300 + 5;
ll N , M , MOD , f[maxn][maxn] , C[maxn][maxn];
ll _s[maxn] , u[maxn];
int main(){
    int x , y , T;
    scanf("%d", &T);
    while( T-- ){
        memset( _s , 0 , sizeof _s );
        memset( f , 0 , sizeof f );
        memset( C , 0 , sizeof C );
        memset( u , 0 , sizeof u );
        scanf("%d %d %d" , &N , &M , &MOD);
        rep( i , 1 , M ){ scanf("%d %d" , &x , &y); u[y]++; } 
        bool flg = 1;
        per( i , N , 1 ){
            _s[i] = _s[i + 1] + u[i];
            if( _s[i] > N - i + 1 ){ flg = 0; puts("NO"); break; }
        }
        if( !flg ) continue;
        rep( i , 0 , N ){
            C[i][0] = C[i][i] = 1;
            rep( j , 1 , i - 1 )
                C[i][j] = ( C[i - 1][j - 1] + C[i - 1][j] ) % MOD;
        }
        f[N + 1][0] = 1;
        per( i , N , 1 )
            rep( j , 0 , N - _s[i] - i + 1 )
                rep( k , 0 , j )
                    f[i][j] = ( f[i][j] + f[i + 1][j - k] * C[j][k] ) % MOD;
        printf("YES %lld\n" , f[1][N - M]);
    }
    return 0;
}