CF1763D Valid Bitonic Permutations

dygxczn

已于 2023-03-22 15:36:16 修改

阅读量83

点赞数 5

文章标签： c++

于 2023-03-21 20:20:37 首次发布

本文链接：https://blog.csdn.net/dygxczn/article/details/129695431

版权

该文阐述了一个编程挑战，涉及找到在一定条件下的位序排列（Bitonicpermutation）数量。问题中给定5个整数n,l,r,x,y，要求找到从1到n的所有数字中，满足特定位置i的值为x，位置j的值为y的升序到降序的位序排列的数量。解决方案包括对x和y的预处理，以及对最高点位置i的枚举，并计算不同区间的组合数。最后，使用组合数学公式计算出答案，并对大整数进行模运算以防止溢出。

摘要由CSDN通过智能技术生成

题目描述

You are given five integers $n$ , $i$ , $j$ , $x$ , and $y$ . Find the number of bitonic permutations $B$ , of the numbers $1$ to $n$ , such that $B_i=x$ , and $B_j=y$ . Since the answer can be large, compute it modulo $10^9+7$ .

A bitonic permutation is a permutation of numbers, such that the elements of the permutation first increase till a certain index $k$ , $\le k \le n-1$ , and then decrease till the end. Refer to notes for further clarification.

题意简述

有五个整数 $n, l, r, x, y$ 和一个长度为 $n$ 的排列 $a$ ，其中 $a_l=x,a_r=y$ 。求有多少个排列满足先升后降。

题解

首先对 $x, y$ 进行处理，若 $x > y$ ，则把整个排列翻转，这样， $x$ 小于 $y$ 方便后面计算。当然， $l, r$ 也要跟着翻转分别变为 $n - l + 1, n - r + 1$ 。

枚举最高点 $n$ 在位置 $i$ ，此时分三种情况考虑：

$i$ 在区间 $(l, r)$ 内，这里放个图方便理解
方便起见，首先考区间 $(i, r)$ 内的方案，区间内有 $n - y - 1$ 个数可以选，有 $r - i - 1$ 个位置可以放，方案数为 $\dbinom{n-y-1}{r-i-1}$
接着考虑区间 $(l, i)$ 内的方案，这时就有些复杂了。由于前面 $n - y - 1$ 个数并没有完全选完，而没选完的这些数只能放在区间 $(l, i)$ 内，它们在区间 $(l, i)$ 内已经占了位置，所以区间 $(l, i)$ 内有 $i-l-1-\left(n-y-1-\left(r-i-1\right)\right)$ 个位置可以放数；有哪些数可以选呢？只能选 $x$ 到 $y$ 之间的数，因为 $y$ 到 $n$ 之间的数已经放好了。方案数即为 $\dbinom{y-x-1}{i-l-1-\left(n-y-1-\left(r-i-1\right)\right)}$
还要考虑区间 $[1, l)$ 的情况，方案数就是 $\dbinom{x-1}{l-1}$
至于区间 $(y, n]$ 就选剩下的，方案数就是 $1$
把方案乘起来即为对答案的贡献。
$i$ 在区间 $(r, n)$ 内，如图

这个情况就比较简单，区间 $(r, i)$ 的方案数为 $\dbinom{n-y-1}{i-r-1}$ ，区间 $(l, r)$ 的方案数为 $\dbinom{y-x-1}{r-l-1}$ ，区间 $[1, l)$ 的方案数为 $\dbinom{x-1}{l-1}$ ，区间 $(i, n]$ 选剩下的。
把方案乘起来即为对答案的贡献。
还有一种情况要特判： $y = n$ ，这种情况要特判，不用枚举最高点。

这种情况也比较简单，区间 $(l, i)$ 的方案数为 $\dbinom{n-x-1}{r-l-1}$ ，区间 $[1, l)$ 的方案数为 $\dbinom{x-1}{l-1}$ ，区间 $(r, n]$ 选剩下的。
对答案的贡献为二者乘积。
这里还有个点很坑，如果 $r = n$ 就不可能出现排列先升后降的情况，应该输出 $0$ 。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=1e9+7;
int n,l,r,x,y;
ll f[1001],inv[1001];
ll ksm(ll a,ll b)
{
    ll ans=1;
    while(b){
        if(b&1) ans=ans*a%mod;
        b>>=1;
        a=a*a%mod;
    }
    return ans;
}
ll c(int n,int m)
{
    if(n<m||n<0||m<0) return 0;
    return f[n]*inv[m]%mod*inv[n-m]%mod;
}
int main()
{
    f[0]=1;
    for(int i=1;i<=1000;i++) f[i]=f[i-1]*i%mod;
    inv[1000]=ksm(f[1000],mod-2);
    for(int i=999;i>=0;i--) inv[i]=inv[i+1]*(i+1)%mod;
    int t;
    cin>>t;
    while(t--){
        cin>>n>>l>>r>>x>>y;
        if(x>y) l=n-l+1,r=n-r+1,swap(x,y),swap(l,r);
        if(y==n){
            if(r==n) puts("0");
            else cout<<c(n-x-1,r-l-1)*c(x-1,l-1)%mod<<endl;
            continue;
        }
        ll ans=0;
        for(int i=l+1;i<r;i++){
            ans=(ans+c(n-y-1,r-i-1)*c(y-x-1,r-l+y-1-n)%mod*c(x-1,l-1))%mod;
        }
        for(int i=r+1;i<n;i++){
            ans=(ans+c(n-y-1,i-r-1)*c(y-x-1,r-l-1)%mod*c(x-1,l-1))%mod;
        }
        cout<<ans<<endl;
    }
}