Number of Ways to Paint N × 3 Grid in C++ program

Suppose we have a grid whose size is n x 3 and we want to paint every cell of the grid with exactly one of the three colors. Here the colors that will be used are Red, Yellow and Green.

Now there is a constraint, that is no two adjacent cells have the same color. We have n number of rows of the grid. Finally, we have to find the number of ways we can paint this grid. The answer may be very large so return it modulo 10^9 + 7.

So, if the input is like 1, then the output will be 12

To solve this, we will follow these steps −

• m = 10^9 + 7

• Define a function add(), this will take a, b,

• return ((a mod m) + (b mod m)) mod m

• From the main method do the following −

• a123 := 6, a121 = 6

• for initialize i := 2, when i −= n, update (increase i by 1), do −

  • b121 := add(3 * a121, 2 * a123)

  • b123 := add(2 * a121, 2 * a123)

  • a121 := b121

  • a123 := b123

• return add(a123, a121)

Example 

Let us see the following implementation to get better understanding −

 Live Demo

#include <bits/stdc++.h>
using namespace std;
typedef long long int lli;
const lli mod = 1e9 + 7;
class Solution {
public:
   lli add(lli a, lli b){
      return ((a % mod) + (b % mod)) % mod;
   }
   int numOfWays(int n){
      lli a123 = 6, a121 = 6;
      lli b123, b121;
      for (int i = 2; i <= n; i++) {
         b121 = add(3 * a121, 2 * a123);
         b123 = add(2 * a121, 2 * a123);
         a121 = b121;
         a123 = b123;
      }
      return add(a123, a121);
   }
};
main(){
   Solution ob;
   cout << (ob.numOfWays(3));
}

Input

3

Output

246

Arnab Chakraborty

Published on 21-Jul-2020 09:14:42

• Related Questions & Answers
• Number of Ways to Paint N × 3 Grid in C++
• Program to find sum of series 1 + 2 + 2 + 3 + 3 + 3 + .. + n in C++
• Program to find sum of series 1*2*3 + 2*3*4+ 3*4*5 + . . . + n*(n+1)*(n+2) in C++
• Ways to paint N paintings such that adjacent paintings don’t have same colors in C++
• How to return a string repeated N number of times in C#?
• Program to find modulus of a number by concatenating n times in Python
• Program to count number of ways we can fill 3 x n box with 2 x 1 dominos in Python
• C/C++ Program to Find the Number Occurring Odd Number of Times?
• Ways to paint N paintings such that adjacent paintings don’t have same colors in C programming
• Program to count number of ways we can throw n dices in Python
• Program to count the number of ways to distribute n number of candies in k number of bags in Python
• Count of different ways to express N as the sum of 1, 3 and 4 in C++
• C++ program to concatenate a string given number of times?
• Maximum number of dots after throwing a dice N times in C++
• C/C++ Program for Finding the Number Occurring Odd Number of Times?