组合计数 ---- 2020 icpc 上海 The Journey of Geor Autumn(思维划分问题计数+预处理优化)

题目链接

---

题目大意：

就是你有一个$n$的全排列，现在问你去重排这个排列使得对于给定的$k$,满足对于任意的$a_i，i>k$的都有$a_i>min(a_{i-1},...,a_{i-k})$
求排列数

---

解题思路：

这里很妙的一点就是我们知道对于最小的数肯定是一定要放到$[1,k]$里面去的，然后最小的数肯定会把数列切成两半，前面那一半是随便选，后面剩下的数又变成新的问题。
那么就是新的问题
$dp[i]表示i个数可以拼出合法的序列个数$
$注意这里是i个数，不一定是i的全排列$
那么$dp[i]={\sum }_{j=1}^k{A}_{i-1}^{j-1}dp[i-j]$
预处理组合数然后前缀和维护$dp$值就行了

---

AC code

#include <bits/stdc++.h>
#define mid ((l + r) >> 1)
#define Lson rt << 1, l , mid
#define Rson rt << 1|1, mid + 1, r
#define ms(a,al) memset(a,al,sizeof(a))
#define log2(a) log(a)/log(2)
#define lowbit(x) ((-x) & x)
#define IOS std::ios::sync_with_stdio(0); cin.tie(0); cout.tie(0)
#define INF 0x3f3f3f3f
#define LLF 0x3f3f3f3f3f3f3f3f
#define f first
#define s second
#define endl '\n'
using namespace std;
const int N = 1e7 + 10, mod = 998244353;
const int maxn = 500010;
const long double eps = 1e-5;
const int EPS = 500 * 500;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> PII;
typedef pair<ll,ll> PLL;
typedef pair<double,double> PDD;
template<typename T> void read(T &x) {
   x = 0;char ch = getchar();ll f = 1;
   while(!isdigit(ch)){if(ch == '-')f*=-1;ch=getchar();}
   while(isdigit(ch)){x = x*10+ch-48;ch=getchar();}x*=f;
}
template<typename T, typename... Args> void read(T &first, Args& ... args) {
   read(first);
   read(args...);
}
int n, k;
inline ll qmi(ll a, ll b) {
    ll res = 1;
    while(b) {
      if(b & 1) res = (res * a) % mod;
      b >>= 1;
      a = a * a % mod;
    }
    return res;
}

ll fac[N], inv[N], sum[N], dp[N];

inline void init() {
  fac[0] = 1;
  for(int i = 1; i < N; ++ i) fac[i] = (fac[i-1] * i) % mod;
  inv[N - 1] = qmi(fac[N-1],mod-2);
  for(int i = N - 2; i >= 0; -- i) inv[i] = inv[i+1] * (i+1) % mod;
}

int main() {
    IOS;
    init();
    cin >> n >> k;
    dp[0] = 1;
    sum[0] = 1;
    for(int i = 1; i <= n; ++ i) {
      dp[i] = (sum[i-1] - (i-k-1>=0?sum[i-k-1]:0) % mod + mod) % mod;
      dp[i] = dp[i] * fac[i-1] % mod;
      sum[i] = (sum[i-1] + (dp[i] * inv[i]) % mod) % mod;
    }
    cout << dp[n] % mod;
    return 0;
}