# Number of ways to insert a character to increase the LCS by one

Given two strings **A** and **B**. The task is to count the number of ways to insert a character in string A to increase the length of the Longest Common Subsequence between string A and string B by 1.**Examples:**

Input : A = “aa”, B = “baaa”

Output : 4

The longest common subsequence shared by string A and string B is “aa”, which has a length of 2.

There are two ways that the length of the longest common subsequence can be increased to 3 by adding a single character to string A:

- There are 3 different positions in string A where we could insert an additional ‘a’ to create longest common subsequence “aaa” (i.e at the beginning, middle, and end of the string).
- We can insert a ‘b’ at the beginning of the string for a new longest common subsequence of “baaa”. So, we have 3 + 1 = 4 ways to insert an alphanumeric character into string A and increase the length of the longest common subsequence by one.

Let say for a given string A and string B, the length of their LCS is **k**. Let’s insert a single character ‘c’ after the ith character in string A and denote the string formed after the insertion as string A_{new}, which looks like:

A_{new} = A_{1, i} . c . A_{i + 1, n}

where A_{i, j} denotes a substring of string A from the ith to the jth characters and ‘.’ denotes a concatenation of two strings.

Let’s define k_{new} to be the length of the LCS of A_{new} and B. Now we want to know if k_{new} = k + 1.

The crucial observation is that the newly inserted character ‘c’ must be a part of any common subsequence of A_{new} and B having length > k. We know this because if there is any common subsequence of A_{new} and B, this is a contradiction because it would mean the length of the LCS of A and B is > k.

Using the above observation, we can try the following approach. For each possible character ‘c'(there are 52 upper and lower case English letters and 10 Arabic digits, so there are 62 possible characters to insert) and for every possible insertion i in String A (there are |a| + 1 insertion positions), let’s try to insert ‘c’ after the ith character in string A and match it with every occurrence of ‘c’ in string B, we can try to match these ‘c’ characters such that:

A_{1, i} . c . A_{i+1, n}

B_{1, j-1} . c . B_{j+1, m}

Now, in order to check if such an insertion produces an LCS of length k + 1, it’s sufficient to check if the length of the LCS of A_{1, i} and B_{1, j-1} plus the length of the LCS A_{i+1, n} and B_{j+1, m} is equal to k. In this case, the lCS of A_{new} and B is k + 1 because there is both a match between the fixed occurrences of character ‘c’ and there is no longer common subsequence between them.

If we can quickly get the length of the LCS between every two prefixes of A and B as well as between every two of their suffixes, we can compute the result. The length of the LCS between their prefixes can be read from a Dynamic Programming table used in computing the LCS of string A and string B. In this method, dp[i][j] stores the length of longest common subsequence of A_{, i} and B_{i, j}. Similarly, the length of the LCS between their suffixes can be read from an analogous dp table which can be computed during computation of the LCS of A_{reversed} and B_{reversed} where S_{reversed} denotes the reversed string S.

## C++

`// CPP Program to Number of ways to insert a` `// character to increase LCS by one` `#include <bits/stdc++.h>` `#define MAX 256` `using` `namespace` `std;` `// Return the Number of ways to insert a` `// character to increase the Longest` `// Common Subsequence by one` `int` `numberofways(string A, string B, ` `int` `N, ` `int` `M)` `{` ` ` `vector<` `int` `> pos[MAX];` ` ` `// Insert all positions of all characters` ` ` `// in string B.` ` ` `for` `(` `int` `i = 0; i < M; i++)` ` ` `pos[B[i]].push_back(i + 1);` ` ` `// Longest Common Subsequence` ` ` `int` `dpl[N + 2][M + 2];` ` ` `memset` `(dpl, 0, ` `sizeof` `(dpl));` ` ` `for` `(` `int` `i = 1; i <= N; i++) {` ` ` `for` `(` `int` `j = 1; j <= M; j++) {` ` ` `if` `(A[i - 1] == B[j - 1])` ` ` `dpl[i][j] = dpl[i - 1][j - 1] + 1;` ` ` `else` ` ` `dpl[i][j] = max(dpl[i - 1][j],` ` ` `dpl[i][j - 1]);` ` ` `}` ` ` `}` ` ` `int` `LCS = dpl[N][M];` ` ` `// Longest Common Subsequence from reverse` ` ` `int` `dpr[N + 2][M + 2];` ` ` `memset` `(dpr, 0, ` `sizeof` `(dpr));` ` ` `for` `(` `int` `i = N; i >= 1; i--) {` ` ` `for` `(` `int` `j = M; j >= 1; j--) {` ` ` `if` `(A[i - 1] == B[j - 1])` ` ` `dpr[i][j] = dpr[i + 1][j + 1] + 1;` ` ` `else` ` ` `dpr[i][j] = max(dpr[i + 1][j],` ` ` `dpr[i][j + 1]);` ` ` `}` ` ` `}` ` ` `// inserting character between position` ` ` `// i and i+1` ` ` `int` `ans = 0;` ` ` `for` `(` `int` `i = 0; i <= N; i++) {` ` ` `for` `(` `int` `j = 0; j < MAX; j++) {` ` ` `for` `(` `auto` `x : pos[j]) {` ` ` `if` `(dpl[i][x - 1] + dpr[i + 1][x + 1] == LCS) {` ` ` `ans++;` ` ` `break` `;` ` ` `}` ` ` `}` ` ` `}` ` ` `}` ` ` `return` `ans;` `}` `// Driver Program` `int` `main()` `{` ` ` `string A = ` `"aa"` `, B = ` `"baaa"` `;` ` ` `int` `N = A.length(), M = B.length();` ` ` `cout << numberofways(A, B, N, M) << endl;` ` ` `return` `0;` `}` |

## Java

`// Java Program for Number of ways to insert a` `// character to increase LCS by one` `import` `java.util.*;` `class` `GFG` `{` ` ` `static` `final` `int` `MAX = ` `256` `;` ` ` `// Return the Number of ways to insert a` ` ` `// character to increase the Longest` ` ` `// Common Subsequence by one` ` ` `static` `int` `numberofways(String A, String B, ` `int` `N, ` `int` `M)` ` ` `{` ` ` `Vector<Integer>[] pos = ` `new` `Vector[MAX];` ` ` `// Insert all positions of all characters` ` ` `// in string B.` ` ` `for` `(` `int` `i = ` `0` `; i < MAX; i++)` ` ` `pos[i] = ` `new` `Vector<>();` ` ` `for` `(` `int` `i = ` `0` `; i < M; i++)` ` ` `pos[B.charAt(i)].add(i + ` `1` `);` ` ` `// Longest Common Subsequence` ` ` `int` `[][] dpl = ` `new` `int` `[N + ` `2` `][M + ` `2` `];` ` ` `for` `(` `int` `i = ` `1` `; i <= N; i++)` ` ` `{` ` ` `for` `(` `int` `j = ` `1` `; j <= M; j++)` ` ` `{` ` ` `if` `(A.charAt(i - ` `1` `) == B.charAt(j - ` `1` `))` ` ` `dpl[i][j] = dpl[i - ` `1` `][j - ` `1` `] + ` `1` `;` ` ` `else` ` ` `dpl[i][j] = Math.max(dpl[i - ` `1` `][j],` ` ` `dpl[i][j - ` `1` `]);` ` ` `}` ` ` `}` ` ` `int` `LCS = dpl[N][M];` ` ` `// Longest Common Subsequence from reverse` ` ` `int` `[][] dpr = ` `new` `int` `[N + ` `2` `][M + ` `2` `];` ` ` `for` `(` `int` `i = N; i >= ` `1` `; i--)` ` ` `{` ` ` `for` `(` `int` `j = M; j >= ` `1` `; j--)` ` ` `{` ` ` `if` `(A.charAt(i - ` `1` `) == B.charAt(j - ` `1` `))` ` ` `dpr[i][j] = dpr[i + ` `1` `][j + ` `1` `] + ` `1` `;` ` ` `else` ` ` `dpr[i][j] = Math.max(dpr[i + ` `1` `][j],` ` ` `dpr[i][j + ` `1` `]);` ` ` `}` ` ` `}` ` ` `// inserting character between position` ` ` `// i and i+1` ` ` `int` `ans = ` `0` `;` ` ` `for` `(` `int` `i = ` `0` `; i <= N; i++)` ` ` `{` ` ` `for` `(` `int` `j = ` `0` `; j < MAX; j++)` ` ` `{` ` ` `for` `(` `int` `x : pos[j])` ` ` `{` ` ` `if` `(dpl[i][x - ` `1` `] +` ` ` `dpr[i + ` `1` `][x + ` `1` `] == LCS)` ` ` `{` ` ` `ans++;` ` ` `break` `;` ` ` `}` ` ` `}` ` ` `}` ` ` `}` ` ` `return` `ans;` ` ` `}` ` ` `// Driver Code` ` ` `public` `static` `void` `main(String[] args)` ` ` `{` ` ` `String A = ` `"aa"` `, B = ` `"baaa"` `;` ` ` `int` `N = A.length(), M = B.length();` ` ` `System.out.println(numberofways(A, B, N, M));` ` ` `}` `}` `// This code is contributed by` `// sanjeev2552` |

## Python3

`# Python Program to Number of ways to insert a` `# character to increase LCS by one` `MAX` `=` `256` `def` `numberofways(A, B, N, M):` ` ` `pos ` `=` `[[] ` `for` `_ ` `in` `range` `(` `MAX` `)]` ` ` `# Insert all positions of all characters` ` ` `# in string B` ` ` `for` `i ` `in` `range` `(M):` ` ` `pos[` `ord` `(B[i])].append(i` `+` `1` `)` ` ` `# Longest Common Subsequence` ` ` `dpl ` `=` `[[` `0` `] ` `*` `(M` `+` `2` `) ` `for` `_ ` `in` `range` `(N` `+` `2` `)]` ` ` `for` `i ` `in` `range` `(` `1` `, N` `+` `1` `):` ` ` `for` `j ` `in` `range` `(` `1` `, M` `+` `1` `):` ` ` `if` `A[i ` `-` `1` `] ` `=` `=` `B[j ` `-` `1` `]:` ` ` `dpl[i][j] ` `=` `dpl[i ` `-` `1` `][j ` `-` `1` `] ` `+` `1` ` ` `else` `:` ` ` `dpl[i][j] ` `=` `max` `(dpl[i ` `-` `1` `][j],` ` ` `dpl[i][j ` `-` `1` `])` ` ` `LCS ` `=` `dpl[N][M]` ` ` `# Longest Common Subsequence from reverse` ` ` `dpr ` `=` `[[` `0` `] ` `*` `(M` `+` `2` `) ` `for` `_ ` `in` `range` `(N` `+` `2` `)]` ` ` `for` `i ` `in` `range` `(N, ` `0` `, ` `-` `1` `):` ` ` `for` `j ` `in` `range` `(M, ` `0` `, ` `-` `1` `):` ` ` `if` `A[i ` `-` `1` `] ` `=` `=` `B[j ` `-` `1` `]:` ` ` `dpr[i][j] ` `=` `dpr[i ` `+` `1` `][j ` `+` `1` `] ` `+` `1` ` ` `else` `:` ` ` `dpr[i][j] ` `=` `max` `(dpr[i ` `+` `1` `][j],` ` ` `dpr[i][j ` `+` `1` `])` ` ` `# inserting character between position` ` ` `# i and i+1` ` ` `ans ` `=` `0` ` ` `for` `i ` `in` `range` `(N` `+` `1` `):` ` ` `for` `j ` `in` `range` `(` `MAX` `):` ` ` `for` `x ` `in` `pos[j]:` ` ` `if` `dpl[i][x ` `-` `1` `] ` `+` `dpr[i ` `+` `1` `][x ` `+` `1` `] ` `=` `=` `LCS:` ` ` `ans ` `+` `=` `1` ` ` `break` ` ` `return` `ans` `# Driver Code` `if` `__name__ ` `=` `=` `"__main__"` `:` ` ` `A ` `=` `"aa"` ` ` `B ` `=` `"baaa"` ` ` `N ` `=` `len` `(A)` ` ` `M ` `=` `len` `(B)` ` ` `print` `(numberofways(A, B, N, M))` `# This code is contributed by vibhu4agarwal` |

## C#

`// C# Program for Number of ways to insert a` `// character to increase LCS by one` `using` `System;` `using` `System.Collections.Generic;` `class` `GFG` `{` ` ` `static` `readonly` `int` `MAX = 256;` ` ` `// Return the Number of ways to insert a` ` ` `// character to increase the longest` ` ` `// Common Subsequence by one` ` ` `static` `int` `numberofways(String A, String B,` ` ` `int` `N, ` `int` `M)` ` ` `{` ` ` `List<` `int` `>[] pos = ` `new` `List<` `int` `>[MAX];` ` ` `// Insert all positions of all characters` ` ` `// in string B.` ` ` `for` `(` `int` `i = 0; i < MAX; i++)` ` ` `pos[i] = ` `new` `List<` `int` `>();` ` ` `for` `(` `int` `i = 0; i < M; i++)` ` ` `pos[B[i]].Add(i + 1);` ` ` `// longest Common Subsequence` ` ` `int` `[,] dpl = ` `new` `int` `[N + 2, M + 2];` ` ` `for` `(` `int` `i = 1; i <= N; i++)` ` ` `{` ` ` `for` `(` `int` `j = 1; j <= M; j++)` ` ` `{` ` ` `if` `(A[i - 1] == B[j - 1])` ` ` `dpl[i, j] = dpl[i - 1, j - 1] + 1;` ` ` `else` ` ` `dpl[i, j] = Math.Max(dpl[i - 1, j],` ` ` `dpl[i, j - 1]);` ` ` `}` ` ` `}` ` ` `int` `LCS = dpl[N, M];` ` ` `// longest Common Subsequence from reverse` ` ` `int` `[,] dpr = ` `new` `int` `[N + 2, M + 2];` ` ` `for` `(` `int` `i = N; i >= 1; i--)` ` ` `{` ` ` `for` `(` `int` `j = M; j >= 1; j--)` ` ` `{` ` ` `if` `(A[i - 1] == B[j - 1])` ` ` `dpr[i, j] = dpr[i + 1, j + 1] + 1;` ` ` `else` ` ` `dpr[i, j] = Math.Max(dpr[i + 1, j],` ` ` `dpr[i, j + 1]);` ` ` `}` ` ` `}` ` ` `// inserting character between position` ` ` `// i and i+1` ` ` `int` `ans = 0;` ` ` `for` `(` `int` `i = 0; i <= N; i++)` ` ` `{` ` ` `for` `(` `int` `j = 0; j < MAX; j++)` ` ` `{` ` ` `foreach` `(` `int` `x ` `in` `pos[j])` ` ` `{` ` ` `if` `(dpl[i, x - 1] +` ` ` `dpr[i + 1, x + 1] == LCS)` ` ` `{` ` ` `ans++;` ` ` `break` `;` ` ` `}` ` ` `}` ` ` `}` ` ` `}` ` ` `return` `ans;` ` ` `}` ` ` `// Driver Code` ` ` `public` `static` `void` `Main(String[] args)` ` ` `{` ` ` `String A = ` `"aa"` `, B = ` `"baaa"` `;` ` ` `int` `N = A.Length, M = B.Length;` ` ` `Console.WriteLine(numberofways(A, B, N, M));` ` ` `}` `}` `// This code is contributed by 29AjayKumar` |

## Javascript

`<script>` `// JavaScript Program for Number of ways to insert a` `// character to increase LCS by one` `let MAX = 256;` `// Return the Number of ways to insert a` ` ` `// character to increase the Longest` ` ` `// Common Subsequence by one` `function` `numberofways(A,B,N,M)` `{` ` ` `let pos = ` `new` `Array(MAX);` ` ` ` ` `// Insert all positions of all characters` ` ` `// in string B.` ` ` `for` `(let i = 0; i < MAX; i++)` ` ` `pos[i] =[];` ` ` ` ` `for` `(let i = 0; i < M; i++)` ` ` `pos[B[i].charCodeAt(0)].push(i + 1);` ` ` ` ` `// Longest Common Subsequence` ` ` `let dpl = ` `new` `Array(N + 2);` ` ` `for` `(let i=0;i<N+2;i++)` ` ` `{` ` ` `dpl[i]=` `new` `Array(M+2);` ` ` `for` `(let j=0;j<M+2;j++)` ` ` `dpl[i][j]=0;` ` ` `}` ` ` ` ` `for` `(let i = 1; i <= N; i++)` ` ` `{` ` ` `for` `(let j = 1; j <= M; j++)` ` ` `{` ` ` `if` `(A[i - 1] == B[j - 1])` ` ` `dpl[i][j] = dpl[i - 1][j - 1] + 1;` ` ` `else` ` ` `dpl[i][j] = Math.max(dpl[i - 1][j],` ` ` `dpl[i][j - 1]);` ` ` `}` ` ` `}` ` ` `let LCS = dpl[N][M];` ` ` ` ` `// Longest Common Subsequence from reverse` ` ` `let dpr = ` `new` `Array(N + 2);` ` ` `for` `(let i=0;i<N+2;i++)` ` ` `{` ` ` `dpr[i]=` `new` `Array(M+2);` ` ` `for` `(let j=0;j<M+2;j++)` ` ` `dpr[i][j]=0;` ` ` `}` ` ` ` ` `for` `(let i = N; i >= 1; i--)` ` ` `{` ` ` `for` `(let j = M; j >= 1; j--)` ` ` `{` ` ` `if` `(A[i - 1] == B[j - 1])` ` ` `dpr[i][j] = dpr[i + 1][j + 1] + 1;` ` ` `else` ` ` `dpr[i][j] = Math.max(dpr[i + 1][j],` ` ` `dpr[i][j + 1]);` ` ` `}` ` ` `}` ` ` ` ` `// inserting character between position` ` ` `// i and i+1` ` ` `let ans = 0;` ` ` `for` `(let i = 0; i <= N; i++)` ` ` `{` ` ` `for` `(let j = 0; j < MAX; j++)` ` ` `{` ` ` `for` `(let x=0;x< pos[j].length;x++)` ` ` `{` ` ` `if` `(dpl[i][pos[j][x] - 1] +` ` ` `dpr[i + 1][pos[j][x] + 1] == LCS)` ` ` `{` ` ` `ans++;` ` ` `break` `;` ` ` `}` ` ` `}` ` ` `}` ` ` `}` ` ` `return` `ans;` `}` `// Driver Code` `let A = ` `"aa"` `, B = ` `"baaa"` `;` `let N = A.length, M = B.length;` `document.write(numberofways(A, B, N, M));` `// This code is contributed by rag2127` `</script>` |

**Output:**

4

**Time Complexity:** O(N x M)