T - Permutation (插入dp)

T - Permutation

题目链接

大致题意:

给出长度为n-1字符串，只包含’’<’‘和’’>’’

求长度是n的全排列(数字为1-n),使得排列的数字大小满足字符串的大于小于关系,问有多少排列方式(mod 1e9+7)

---

解题思路:

状态表示:

f[i][j]表示前i个数字为1-i的全排列并且第i个数字是j且满足大小关系的方案数

sum[i][j]表示第i个位置放1-j的方案数之和,即f数组的前缀和

分析:

当s[i]==’<’ 且第i+1个位置是数字j时,我们需要找前一个数字小于j且满足前面字符串关系的方案数

即f[i+1][j]=${\sum }_{k=1}^{k=j-1}f[i][k]$

当s[i]==’>’ 且第i+1个位置是数字j时,我们需要找前一个数字大于等于j且不超过i+1且满足前面字符串关系的方案数

即f[i+1][j]=${\sum }_{k=j}^{k=i+1}f[i][k]$

解释为什么从大于等于j开始,我们怎么确定是1到i+1的全排列呢?当我们在第i+1位置插入j时,使前i-1个数字中,大于等于j的数字+1即可,必定保证这是全排列

转移方程:

用sum数组进行前缀和优化,得

s[i]==’<’ f[i + 1][j] = (f[i + 1][j] + sum[i][j - 1]) % mod

s[i]==’>’ f[i + 1][j] = (f[i + 1][j] + sum[i][i + 1] - sum[i][j - 1] + mod) % mod

补充:类似题目 F - Deforestation 题解

---

AC代码:

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
#define debug(a) cout << #a << " = " << a << endl;
using namespace std;
typedef long long ll;
const int mod = 1e9 + 7;
const int N = 3100;
int n, m;
char s[N];
ll f[N][N]; //表示前i个数字为1-i的全排列并且第i个数字是j且满足大小关系的方案数
ll sum[N][N];//表示第i个位置放1-j的方案数之和
int main(void)
{
	ios::sync_with_stdio(0); cin.tie(0);
	cin >> n;
	cin >> s + 1;

	f[1][1] = 1;
	for (int i = 1; i <= n; ++i)sum[1][i] = 1;
	for (int i = 1; i <= n - 1; ++i) {
		for (int j = 1; j <= i + 1; ++j) {
			if (s[i] == '<')f[i + 1][j] = (f[i + 1][j] + sum[i][j - 1]) % mod;
			else f[i + 1][j] = (f[i + 1][j] + sum[i][i + 1] - sum[i][j - 1] + mod) % mod;
		}
		for (int j = 1; j <= n; ++j)
			sum[i + 1][j] = (sum[i + 1][j - 1] + f[i + 1][j]) % mod;
	}

	ll res = 0;
	for (int i = 1; i <= n; ++i) res = (res + f[n][i]) % mod;
	cout << res << endl;
	return 0;
}

---