Atcoder agc024E

最新推荐文章于 2020-04-06 23:31:28 发布

mayaohua2003

最新推荐文章于 2020-04-06 23:31:28 发布

阅读量274

点赞数

分类专栏：集训队作业 atcoder 动态规划

本文链接：https://blog.csdn.net/qq_38609262/article/details/104670772

版权

集训队作业同时被 3 个专栏收录

141 篇文章 3 订阅

订阅专栏

atcoder

63 篇文章 0 订阅

订阅专栏

动态规划

47 篇文章 1 订阅

订阅专栏

博客详细探讨了Atcoder AGC024E问题的解决方案，重点在于如何从给定序列中删除字符，使得相邻字符不相等，并计算不同删除方案的数量。通过设定F[i][j]表示长度为i且元素在1到j之间的序列的不同删除方案，利用组合数进行状态转移，避免了四次方的时间复杂度，实现了O(N²K)的时间复杂度优化。

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

倒着考虑，从一个给定的 $A_N$ 不断删除字符能得到多少种不同的方案。为了不计重复，我们要求每次删掉一个字符 $x_i$ 时，满足 $x_{i+1}\neq x_i$ ，那么合法的条件是 $\exist j\geq i,x_i=x_{i+1}=...=x_j,x_{j+1}<x_i$ （假设最后补一个 $0$ ）。
考虑计数，设 $F [i] [j]$ 表示当前序列长度为 $i$ ，元素在 $1$ 到 $j$ 之间，枚举最后一个删掉的位置为 $k$ ，权值为 $l$ ，那么 $1$ 到 $k - 1$ 间的元素均大于 $l$ ，而 $k + 1$ 到 $i$ 间无限制，两部分互相独立，可以用组合数合并。直接转移是四次方的，但注意到 $l$ 与后半部分无关，可以预处理转移加速。
时间复杂度 $\mathcal O(N^2K)$ 。

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

int MOD;

ll C[305][305];
int f[305][305],sum[305][305];

void pre(int n) {
  for(int i=0;i<=n;i++) C[i][0]=1;
  for(int i=1;i<=n;i++)
    for(int j=1;j<=i;j++) C[i][j]=(C[i-1][j-1]+C[i-1][j])%MOD;
}

int main() {
  int n,k;
  scanf("%d%d%d",&n,&k,&MOD);
  pre(n);
  for(int i=1;i<=k+1;i++) {
  	f[0][i]=1;
  	sum[0][i]=i;
  }
  for(int i=1;i<=n;i++) {
  	for(int j=1;j<=k+1;j++)
  	  for(int l=1;l<=i;l++) 
		f[i][j]=(f[i][j]+(ll)sum[l-1][j-1]*f[i-l][j]%MOD*C[i-1][l-1])%MOD;
  	for(int j=1;j<=k+1;j++) sum[i][j]=(sum[i][j-1]+f[i][j])%MOD;
  }
  printf("%d\n",f[n][k+1]);
  return 0;
}