洛谷P5241 序列

Address

洛谷P5241

Solution

• 考虑构造出一条 $n$ 个点的有向链，那么每次就可以将任意个数的点缩成一个。
• 因此我们可以将每一个本质不同的序列 $B$ 与每加一条边都进行下面的其中一个操作一一对应：

1. 在有向链的末尾增加一个点；
2. 将有向链的前若干个点缩成一个点；
3. 若有向链包含的点数已经为 $n$，任加一条不会将点缩起来的边。

• 对于其余的情形，因为在所需边数以及缩点的范围方面都不如直接构造有向链要优，可以不考虑。
• 注意到操作 1 和操作 3 不会交替出现，因此可以将这两个过程分开 DP。
• 对于只进行操作 1 和操作 2 的过程，设 $f_{i,j,k}(k\le j)$ 表示已经加了 $i$ 条边、有向链包含的点数为 $j$、前 $k$ 个点已经被缩成一个点时的方案数， 易得转移：
  $$\begin{array}{r}{f}_{i,j,k}=f_{i-1,j-1,k}+\sum _{r=1}^{k-1}{f}_{i-1,j,r}\end{array}$$
• 对于只进行操作 2 和操作 3 的过程，设 $g_{i,k}$ 表示已经加了 $i$ 条边、前 $k$ 个点已经被缩成一个点时的方案数，易得转移：
  $$\begin{array}{r}{g}_{i,k}=f_{i,n,k}+\sum _{r=1}^k{g}_{i-1,r}\end{array}$$
• 注意到操作 3 可以进行的次数不是无限的，因此我们需要限制状态中的 $i$ 不能超过将前 $k$ 个点缩成一个点时图中可能的最大边数，即：
  $$\begin{array}{r}i\le k(n-1)+\frac{(n-k)(n-k-1)}{2}\end{array}$$
• 最后加入 $i$ 条边的答案为：
  $$\begin{array}{r}an{s}_i=\sum _{j=1}^{n-1}\sum _{k=1}^j{f}_{i,j,k}+\sum _{k=1}^n{g}_{i,k}\end{array}$$
• 前缀和优化即可，显然 $f$ 中的 $i$ 只要枚举到 $2(n-1)$，总的时间复杂度 $\mathcal{O}(n^3)$。

Code

#include <bits/stdc++.h>

template <class T>
inline void read(T &res)
{
	char ch;
	while (ch = getchar(), !isdigit(ch));
	res = ch ^ 48;
	while (ch = getchar(), isdigit(ch))
		res = res * 10 + ch - 48;
}

template <class T>
inline void put(T x)
{
	if (x > 9)
		put(x / 10);
	putchar(x % 10 + 48);
}

const int N = 4e2 + 5;
const int M = 16e4 + 5;
const int mod = 1e9 + 7;
int f[2][N][N], sf[2][N][N], g[M][N], ans[M], n, m;

template <class T>
inline T Min(T x, T y) {return x < y ? x : y;}

inline void add(int &x, int y)
{
	x += y;
	x >= mod ? x -= mod : 0;
}

int main()
{
	read(n);
	if (n <= 1)
		return 0;

	m = n * (n - 1);
	f[0][1][1] = 1;
	for (int k = 1; k <= n; ++k)
		sf[0][1][k] = 1;
	for (int i = 1, im = n - 1 << 1; i <= im; ++i)
	{
		int now = i & 1, lst = now ^ 1;
		for (int j = 1; j <= n; ++j)
			for (int k = 1; k <= j; ++k)
				add(f[now][j][k] = f[lst][j - 1][k], sf[lst][j][k - 1]);
		for (int k = 1; k <= n; ++k)
			add(g[i][k], f[lst][n - 1][k]);
		for (int j = 1; j <= n; ++j)
		{
			for (int k = 1; k <= n; ++k)
			{
				add(sf[now][j][k] = f[now][j][k], sf[now][j][k - 1]);
				if (j < n)
					add(ans[i], f[now][j][k]);
			}
		}
	}
	for (int i = n - 1; i <= m; ++i)
	{
		int res = 0;
		for (int j = 1; j <= n; ++j)
		{
			add(res, g[i - 1][j]);
			if (i <= j * (n - 1) + (n - j) * (n - j - 1) / 2)
				add(g[i][j], res);
			add(ans[i], g[i][j]);
		}
	}

	for (int i = 1; i <= m; ++i)
		put(ans[i]), putchar(' ');
	return 0;
}