Given three positive integers n, k and x. The task is to count the number of different array that can be formed of size n such that each element is between 1 to k and two consecutive element are different. Also, the first and last elements of each array should be 1 and x respectively.
Input : n = 4, k = 3, x = 2 Output : 3
The idea is to use Dynamic Programming and combinatorics to solve the problem.
First of all, notice that the answer is same for all x from 2 to k. It can easily be proved. This will be useful later on.
Let the state f(i) denote the number of ways to fill the range [1, i] of array A such that A1 = 1 and Ai ≠ 1.
Therefore, if x ≠ 1, the answer to the problem is f(n)/(k – 1), because f(n) is the number of way where An is filled with a number from 2 to k, and the answer are equal for all such values An, so the answer for an individual value is f(n)/(k – 1).
Otherwise, if x = 1, the answer is f(n – 1), because An – 1 ≠ 1, and the only number we can fill An with is x = 1.
Now, the main problem is how to calculate f(i). Consider all numbers that Ai – 1 can be. We know that it must lie in [1, k].
- If Ai – 1 ≠ 1, then there are (k – 2)f(i – 1) ways to fill in the rest of the array, because Ai cannot be 1 or Ai – 1 (so we multiply with (k – 2)), and for the range [1, i – 1], there are, recursively, f(i – 1) ways.
- If Ai – 1 = 1, then there are (k – 1)f(i – 2) ways to fill in the rest of the array, because Ai – 1 = 1 means Ai – 2 ≠ 1 which means there are f(i – 2)ways to fill in the range [1, i – 2] and the only value that Ai cannot be 1, so we have (k – 1) choices for Ai.
By combining the above, we get
f(i) = (k - 1) * f(i - 2) + (k - 2) * f(i - 1)
This will help us to use dynamic programming using f(i).
Below is the implementation of this approach:
Attention reader! Don’t stop learning now. Get hold of all the important DSA concepts with the DSA Self Paced Course at a student-friendly price and become industry ready.
- Count of binary strings of length N having equal count of 0's and 1's and count of 1's ≥ count of 0's in each prefix substring
- Count of Binary strings of length N having atmost M consecutive 1s or 0s alternatively exactly K times
- Count of N digit Numbers having no pair of equal consecutive Digits
- Count numbers in a range with digit sum divisible by K having first and last digit different
- Count of all subsequences having adjacent elements with different parity
- Generate a Binary String without any consecutive 0's and at most K consecutive 1's
- Count of binary strings of length N with even set bit count and at most K consecutive 1s
- Count of possible arrays from prefix-sum and suffix-sum arrays
- Maximum sub-matrix area having count of 1's one more than count of 0's
- Count of binary strings of length N having equal count of 0's and 1's
- Find longest bitonic sequence such that increasing and decreasing parts are from two different arrays
- Number of ways to select equal sized subarrays from two arrays having atleast K equal pairs of elements
- Number of sub-sequence such that it has one consecutive element with difference less than or equal to 1
- Count number of binary strings without consecutive 1's
- Count Pairs Of Consecutive Zeros
- Count ways to reach a score using 1 and 2 with no consecutive 2s
- Count of N-digit numbers in base K with no two consecutive zeroes
- Count of strings possible by replacing two consecutive same character with new character
- Count number of binary strings without consecutive 1’s : Set 2
- Count the number of ordered sets not containing consecutive numbers
If you like GeeksforGeeks and would like to contribute, you can also write an article using contribute.geeksforgeeks.org or mail your article to firstname.lastname@example.org. See your article appearing on the GeeksforGeeks main page and help other Geeks.
Please Improve this article if you find anything incorrect by clicking on the "Improve Article" button below.
Improved By : vt_m