Rotate Array Right By K Positions

Write a program to rotate array by K Positions.

Example:

Input:  arr[] = {1,2,3,4,5,6,7}, k = 3
Output: [5,6,7,1,2,3,4]

Approach

C++

#include <bits/stdc++.h>
using namespace std;

void reverseArray(vector<int> &arr, int low, int high)
{
    while (low < high)
    {
        swap(arr[low], arr[high]);
        low++;
        high--;
    }
}
void rotateArrayRightByK(vector<int> &arr, int k)
{

    //reverse last k elements
    reverseArray(arr, arr.size() - k, arr.size() - 1);

    //reverse the firt n-k elements
    reverseArray(arr, 0, arr.size() - k - 1);

    //reverse the whole array
    reverseArray(arr, 0, arr.size() - 1);
}
int main()
{
    vector<int> arr = {1, 2, 3, 4, 5, 6, 7};

    int k = 3;

    k = k % arr.size();
    rotateArrayRightByK(arr, k);
    cout << "[";
    for (int i = 0; i < arr.size(); i++)
    {
        cout << arr[i];
        if (i != arr.size() - 1)
            cout << ",";
    }
    cout << "]";

    return 0;
}


No comments:

Post a Comment