题目
题意:
计算长度为 n 的所有排列中所有最大值为 n 且最小值为 1 的子区间个数,对 998244353 取模。
思路: 看1和n之间的距离是1、2、3,然后找规律。当距离是k时,稳定有(n-k)的贡献,之后看起点1的位置,左边有i-1个数,右边有n-(i+k)个数,可以两两组合,直接相乘。所以就能推出求和公式了。
PS: 这里只考虑了1和n的关系,还可以倒过来,乘2,然后其他数是可以任意排列的,还有(n-2)!的贡献.
还可以进一步化简
题解说能化简成O(1),上述式子可以再拆开,然后套立方和、平方和公式啥的。
套公式即可。
代码:
#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int> PII;
typedef long long ll;
const int N = 1e6+10;
const int mod = 998244353;
ll n;
int m,k,T;
ll fac[N];
ll qpow(ll a,int k)
{
ll res = 1;
while(k)
{
if(k&1) res = res * a % mod;
a = a * a % mod;
k >>= 1;
}
return res;
}
ll ni(ll x)
{
return qpow(x,mod-2);
}
void solve()
{
cin>>n;
if(n==1)
{
cout<<1; return ;
}
ll ans = 0;
ll fenmu1 = ni(2);
ll fenmu2 = ni(6);
for(ll k=1;k<=n-1;++k)
{
ll t = (n-k)*(n-k+1)%mod*(n-k+2)%mod*fenmu2%mod;
ans = (ans + t) % mod;
// ll t1 = 1ll*(n-k+1)*(n-k)%mod*(n-k+1)%mod*fenmu1%mod;
// ll t2 = 1ll*(n-k)*(n-k+1)%mod*(2*n-2*k+1)%mod*fenmu2%mod;
// ans = (ans + t1) % mod;
// ans = (ans - t2) % mod;
}
ans %= mod; ans = (ans + mod) % mod;
ans = ans * fac[n-2] % mod;
ans = ans * 2 % mod;
cout<<ans;
}
signed main(void)
{
fac[0] = 1;
for(int i=1;i<N;++i) fac[i] = fac[i-1] * i % mod;
T = 1;
// cin>>T;
while(T--)
solve();
return 0;
}
/*
3
2 2 2
2 2 1
*/