Circular Array Rotation

John Watson knows of an operation called a right circular rotation on an array of integers. One rotation operation moves the last array element to the first position and shifts all remaining elements right one. To test Sherlock's abilities, Watson provides Sherlock with an array of integers. Sherlock is to perform the rotation operation a number of times then determine the value of the element at a given position.

For each array, perform a number of right circular rotations and return the values of the elements at the given indices. 

Example:

Input:  n=3,k=2,q=3, arr[]={1,2,3},queries={0,1,2}
Output: 2
3
1

Approach

Java


import java.util.Arrays;

public class CircularArrayRotation {
    public static void main(String[] args) {
        int k = 2;
        int[] arr = { 123 };
        int[] queries = { 012 };
        int[] res = circularArrayRotation(arr, k, queries);
        System.out.println(Arrays.toString(res));
    }

    static int[] circularArrayRotation(int[] arrint kint[] queries) {
        int n = arr.length;
        int[] res = new int[queries.length];
        k = k % n;
        int temp[] = new int[n - k];
        for (int i = 0; i < n - k; i++) {
            temp[i] = arr[i];
        }
        int l = 0;
        for (int i = n - k; i < n; i++) {
            arr[l++] = arr[i];
        }
        for (int i = 0; i < n - k; i++) {
            arr[l++] = temp[i];
        }
        for (int i = 0; i < queries.length; i++) {
            int j = queries[i];
            res[i] = arr[j];
        }
        return res;
    }
}

C++

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

vector<intcircularArrayRotation(vector<intarrint kvector<intqueries)
{
    vector<intres;
    int n = arr.size();
    k = k % n;
    int temp[n - k];
    for (int i = 0i < n - ki++)
    {
        temp[i] = arr[i];
    }
    int l = 0;
    for (int i = n - ki < ni++)
    {
        arr[l++] = arr[i];
    }
    for (int i = 0i < n - ki++)
    {
        arr[l++] = temp[i];
    }
    for (int i = 0i < ni++)
    {
        int j = queries[i];
        res.push_back(arr[j]);
    }
    return res;
}
int main()
{
    int n = 3k = 2q = 3;
    vector<intarr = {123};
    vector<intqueries = {012};
    vector<intres = circularArrayRotation(arrkqueries);
    for (int i = 0i < res.size(); i++)
    {
        cout << res[i] << "\n";
    }
    return 0;
}


No comments:

Post a Comment