牛客小白月赛83-D-小天的子序列

最新推荐文章于 2024-10-17 11:39:06 发布

随心度春秋

最新推荐文章于 2024-10-17 11:39:06 发布

阅读量149

点赞数 1

文章标签：算法动态规划

本文链接：https://blog.csdn.net/aiwangyi6/article/details/136506463

版权

文章讲述了在NowCoder竞赛中解决小天的子序列问题时，如何利用数论中的Lucas定理和快速幂计算组合数，进行预处理以实现O(1)查询，强调了仔细审题和正确理解状态转移的重要性。

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

dp+数论的一道题

这题最开始想复杂了,考虑组合可能存在重复的情况,怎么去重,后面再读一遍题发现题目已经规定了不考虑重复,所以还是要认真审题

状态其实还是很好想的

f[开头][结尾][长度]

不过这个dp严格意义上说压根不是dp,因为压根没有递推的过程,纯粹的预处理,这里写成dp主要是理解的时候当dp了,后面的思路也在考虑dp

数论方面就是Lucas定理求组合数

预处理然后O(1)查询就行,不会卡常,还是很友好的

// Problem: 小天的子序列
// Contest: NowCoder
// URL: https://ac.nowcoder.com/acm/contest/72041/D
// Memory Limit: 262144 MB
// Time Limit: 2000 ms
// Date: 2024-03-06 12:33:30
// 
// Powered by CP Editor (https://cpeditor.org)

#include<bits/stdc++.h>
#include<unordered_map>
#define endl '\n'
#define int int64_t
using namespace std;
const int mod = 998244353;
int f[30][30][505], rec[505],n;
const int N = 505;
int f1[N], g[N];
int qpow(int a, int b, int p) {
	int res = 1;
	while (b) {
		if (b & 1)res = res * a % p;
		a = a * a % p, b >>= 1;
	}
	return res;
}
void init(int n, int p) {
	memset(f, 0, sizeof f);
	memset(g, 0, sizeof g);
	f1[0] = g[0] = 1;
	for (int i = 1; i <= n; ++i) {
		f1[i] = i * f1[i - 1] % p;
		g[i] = g[i - 1] * qpow(i, p - 2, p) % p;
	}
}
int get(int n, int m, int p) {
	if (n < m) return 0;
	return f1[n] * g[m] % p * g[n - m] % p;
}//上面的就是乘法逆元快速幂求组合数的预处理
int Lucas(int n, int m, int p) {
	if (m == 0)return 1;
	return Lucas(n / p, m / p, p) * get(n % p, m % p, p);
}//这就是卢卡斯定理
void solve() {
	cin >> n;
	string s; cin >> s;
	init(n, mod);
	for (int i = 0; i < n; ++i) {
		for (int j = i + 1; j < n; ++j) {
			for (int len = 2; len <= j - i + 1; ++len) {
				f[s[i] - 'a'][s[j] - 'a'][len] = (f[s[i] - 'a'][s[j] - 'a'][len] +
					Lucas(j - i - 1,len - 2,mod)) % mod;
			}
		}
	}
	int m; cin >> m;
	for (int i = 1; i <= m; ++i) {
		char ch1, ch2; cin >> ch1 >> ch2;
		int len; cin >> len;
		cout << f[ch1 - 'a'][ch2 - 'a'][len] << endl;
	}
}
signed main() {
	ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
	int t = 1;
	while (t--) {
		solve();
	}
	return 0;
}