pub fn number_of_common_subsequences<T: Eq>(
modulus: i64,
a: &[T],
b: &[T],
) -> i64 {
let n = a.len();
let m = b.len();
let mut dp = vec![vec![1i64; m + 1]; n + 1];
for i in 0..n {
for j in 0..m {
dp[i + 1][j + 1] = dp[i + 1][j] + dp[i][j + 1];
if a[i] != b[j] {
dp[i + 1][j + 1] -= dp[i][j];
}
dp[i + 1][j + 1] %= modulus;
}
}
(dp[n][m] + modulus) % modulus
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test() {
let m = 1_000_000_007;
let cases = vec![
((vec![1, 3], vec![3, 1]), 3),
((vec![1, 1], vec![1, 1]), 6),
((vec![3, 4, 5, 6], vec![3, 4, 5, 6]), 16),
(
(
vec![9, 6, 5, 7, 5, 9, 8, 5, 6, 7],
vec![8, 6, 8, 5, 5, 7, 9, 9, 7],
),
191,
),
(
(
vec![
1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1, 1,
],
vec![
1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1, 1,
],
),
846527861,
),
];
for ((a, b), ans) in cases {
assert_eq!(number_of_common_subsequences(m, &a, &b), ans);
}
}
}