2021牛客寒假算法基础集训营5 B.比武招亲（上）

题目链接

思路

这题多列举一些样例会发现，可取的情况数量与差值实际为排列组合关系。比如当前有n个数，一次选取m个，差值的范围显然为1至n-1。而对于每种差值，我们可以确定它的前两位，比如差值为1的开头两位必定是相邻的两个数，此时如何确定组合数呢？为了不改变当前确定的差值，即不改变序列中的最大值与最小值，所以在当前差值所取得的两数之间的数都是可取的，假设当前枚举的差值为i，我们可以从i+1(包括当前取得的两个数，比如差值为1，那么可取的数为差值为1的两个数本身)个数中任意选取m-2(初始两个已确定)个出来，组成符合条件的序列。而上述的在n个数中取出m个，且可存在重复的数量为C(n+m-1,m),当然，此时的结果仅对应差值为i时的某一种情况，而情况总数则为n-i(写一下就明白了)，然后再乘上当前的差值(即贡献)累加即可。

代码

#include<bits/stdc++.h>
using namespace std;
const int P=998244353;
const int maxn=1e6+5;
typedef long long ll;
const int inf=1e9;
ll m,n,ans,jc[maxn];
void cjc()
{
    jc[0]=jc[1]=1;
    for(ll i=2;i<maxn;i++)
        jc[i]=jc[i-1]*i%P;
}
ll ksm(ll a,ll b)
{
    ll ans = 1;
    while(b)
    {
        if(b&1)
            ans=ans*a%P;
        a=a*a%P;
        b>>=1;
    }
    return ans%P;
}
ll inverse(ll a)//费马小定理求逆元
{
    return ksm(a,P-2);
}
ll C(ll n,ll m)
{
    return jc[n]*inverse(jc[m])%P*inverse(jc[n-m])%P;
}//组合数取模
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
    cjc();
    cin>>n>>m;
    m-=2;//可被填充的位置数
    for(int i=1;i<=n-1;i++)
    {
        ll tmp=i*(n-i)%P;//i为贡献，n-i为情况总数
        ll res=i+1;//可用于填充的数的数量
        ans=(ans+tmp*C(res+m-1,m)%P)%P;
    }//枚举差值
    cout<<ans<<endl;
    return 0;
}