Shift matrix elements row-wise by k

Read

Discuss

Given a square matrix mat[][] and a number k. The task is to shift the first k elements of each row to the right of the matrix.

Examples :

Input : mat[N][N] = {{1, 2, 3},
                     {4, 5, 6},
                     {7, 8, 9}}
        k = 2
Output :mat[N][N] = {{3, 1, 2}
                     {6, 4, 5}
                     {9, 7, 8}}

Input : mat[N][N] = {{1, 2, 3, 4}
                     {5, 6, 7, 8}
                     {9, 10, 11, 12}
                     {13, 14, 15, 16}}
        k = 2
Output :mat[N][N] = {{3, 4, 1, 2}
                     {7, 8, 5, 6}
                     {11, 12, 9, 10}
                     {15, 16, 13, 14}}
Note: Matrix should be a square matrix

Implementation:

C++

// C++ program to shift k elements in a matrix. 
#include <bits/stdc++.h> 
using namespace std; 
#define N 4 
  
// Function to shift first k elements of  
// each row of matrix. 
void shiftMatrixByK(int mat[N][N], int k) 
{ 
    if (k > N) { 
        cout << "shifting is not possible" << endl; 
        return; 
    } 
      
    int j = 0; 
    while (j < N) { 
          
        // Print elements from index k 
        for (int i = k; i < N; i++) 
            cout << mat[j][i] << " "; 
              
        // Print elements before index k 
        for (int i = 0; i < k; i++) 
            cout << mat[j][i] << " "; 
              
        cout << endl; 
        j++; 
    } 
} 
  
// Driver code 
int main() 
{ 
    int mat[N][N] = {{1, 2, 3, 4}, 
                     {5, 6, 7, 8}, 
                     {9, 10, 11, 12}, 
                     {13, 14, 15, 16}}; 
    int k = 2; 
      
    // Function call 
    shiftMatrixByK(mat, k); 
      
    return 0; 
} 

Java

// Java program to shift k elements in a  
// matrix. 
import java.io.*; 
import java.util.*; 
  
public class GFG { 
       
    static int N = 4; 
       
    // Function to shift first k elements  
    // of each row of matrix. 
    static void shiftMatrixByK(int [][]mat, 
                                    int k) 
    { 
        if (k > N) { 
            System.out.print("Shifting is"
                        + " not possible"); 
            return; 
        } 
           
        int j = 0; 
        while (j < N) { 
               
            // Print elements from index k 
            for (int i = k; i < N; i++) 
                System.out.print(mat[j][i] + " "); 
                   
            // Print elements before index k 
            for (int i = 0; i < k; i++) 
                System.out.print(mat[j][i] + " "); 
                   
            System.out.println(); 
            j++; 
        } 
    } 
       
    // Driver code 
    public static void main(String args[]) 
    { 
        int [][]mat = new int [][] 
                    { {1, 2, 3, 4}, 
                      {5, 6, 7, 8}, 
                      {9, 10, 11, 12}, 
                      {13, 14, 15, 16} }; 
        int k = 2; 
           
        // Function call 
        shiftMatrixByK(mat, k); 
    } 
} 
   
// This code is contributed by Manish Shaw  
// (manishshaw1) 

Python3

# Python3 program to shift k  
# elements in a matrix. 
  
N = 4
# Function to shift first k  
# elements of each row of  
# matrix. 
def shiftMatrixByK(mat, k): 
    if (k > N) : 
        print ("shifting is"
            " not possible") 
        return
      
    j = 0
    while (j < N) : 
          
        # Print elements from 
        # index k 
        for i in range(k, N): 
            print ("{} " .  
            format(mat[j][i]), end="") 
              
        # Print elements before 
        # index k 
        for i in range(0, k): 
            print ("{} " .  
            format(mat[j][i]), end="") 
              
        print ("") 
        j = j + 1
  
# Driver code 
mat = [[1, 2, 3, 4], 
       [5, 6, 7, 8], 
       [9, 10, 11, 12], 
       [13, 14, 15, 16]] 
k = 2
  
# Function call 
shiftMatrixByK(mat, k) 
  
# This code is contributed by  
# Manish Shaw (manishshaw1) 

C#

// C# program to shift k elements in a  
// matrix. 
using System; 
  
class GFG { 
      
    static int N = 4; 
      
    // Function to shift first k elements  
    // of each row of matrix. 
    static void shiftMatrixByK(int [,]mat, 
                                    int k) 
    { 
        if (k > N) { 
            Console.WriteLine("shifting is"
                        + " not possible"); 
            return; 
        } 
          
        int j = 0; 
        while (j < N) { 
              
            // Print elements from index k 
            for (int i = k; i < N; i++) 
                Console.Write(mat[j,i] + " "); 
                  
            // Print elements before index k 
            for (int i = 0; i < k; i++) 
                Console.Write(mat[j,i] + " "); 
                  
            Console.WriteLine(); 
            j++; 
        } 
    } 
      
    // Driver code 
    public static void Main() 
    { 
        int [,]mat = new int [,] 
                    { {1, 2, 3, 4}, 
                      {5, 6, 7, 8}, 
                      {9, 10, 11, 12}, 
                      {13, 14, 15, 16} }; 
        int k = 2; 
          
        // Function call 
        shiftMatrixByK(mat, k); 
    } 
} 
  
// This code is contributed by Manish Shaw  
// (manishshaw1) 

PHP

<?php 
// PHP program to shift k  
// elements in a matrix. 
  
// Function to shift first k  
// elements of each row of matrix. 
function shiftMatrixByK($mat, $k) 
{ 
    $N = 4; 
    if ($k > $N)  
    { 
        echo ("shifting is not possible\n"); 
        return; 
    } 
      
    $j = 0; 
    while ($j < $N)  
    { 
          
        // Print elements from index k 
        for ($i = $k; $i < $N; $i++) 
            echo ($mat[$j][$i]." "); 
              
        // Print elements before index k 
        for ($i = 0; $i < $k; $i++) 
            echo ($mat[$j][$i]." "); 
              
        echo ("\n"); 
        $j++; 
    } 
} 
  
// Driver code 
$mat = array(array(1, 2, 3, 4), 
             array(5, 6, 7, 8), 
             array(9, 10, 11, 12), 
             array(13, 14, 15, 16)); 
$k = 2; 
      
// Function call 
shiftMatrixByK($mat, $k); 
  
// This code is contributed by  
// Manish Shaw(manishshaw1) 
?> 

Javascript

<script> 
// Java Script program to shift k elements in a 
// matrix. 
let N = 4; 
      
// Function to shift first k elements 
// of each row of matrix. 
function shiftMatrixByK(mat,k) 
{ 
    if (k > N) { 
        document.write("Shifting is not possible"); 
        return; 
    } 
          
    let j = 0; 
    while (j < N) { 
              
        // Print elements from index k 
        for (let i = k; i < N; i++) 
            document.write(mat[j][i] + " "); 
                  
        // Print elements before index k 
        for (let i = 0; i < k; i++) 
            document.write(mat[j][i] + " "); 
                  
        document.write("<br>"); 
        j++; 
    } 
} 
      
// Driver code 
      
    let mat =  
                [[1, 2, 3, 4], 
                [5, 6, 7, 8], 
                [9, 10, 11, 12], 
                [13, 14, 15, 16]]; 
    let k = 2; 
          
    // Function call 
    shiftMatrixByK(mat, k); 
      
      
// This code is contributed by sravan kumar G 
</script>

Output

3 4 1 2 
7 8 5 6 
11 12 9 10 
15 16 13 14

Complexity Analysis:

Time Complexity: O(n²),
Auxiliary Space: O(1)

Feeling lost in the world of random DSA topics, wasting time without progress? It's time for a change! Join our DSA course, where we'll guide you on an exciting journey to master DSA efficiently and on schedule.
Ready to dive in? Explore our Free Demo Content and join our DSA course, trusted by over 100,000 geeks!

Last Updated : 20 Feb, 2023