c_cpp 人力资源阵列左转

Posted 2021-05-14

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

// #HR #InterviewPreparationKit #Arrays #Problem
// https://www.hackerrank.com/challenges/ctci-array-left-rotation/problem?h_l=playlist&slugs%5B%5D=interview&slugs%5B%5D=interview-preparation-kit&slugs%5B%5D=arrays

vector<int> rotLeft(vector<int> a, int d) {
    int n=a.size();
    int r=n-d; // right rotations reqd = size-left rotations
    vector<int> ans(n);
    for(int i=0;i<n;i++){
        ans[(i+r)%n]=a[i];  // logic for right rotation
    }
    return ans;

}

int main(){
    int n;
    cin>>n;
    vector<int> a(n);
    for(int i=0;i<n;i++){
        cin>>a[i];
    }
    int d;
    cin>>d;
    vector<int> arot=rotLeft(a,d);
    for(int i=0;i<n;i++){
        cout<<a[i]<<" ";
    }
    cout<<endl;
    for(int i=0;i<n;i++){
        cout<<arot[i]<<" ";
    }
    cout<<endl;
    return 0;
}