Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> C++

Xoay mảng sang trái K vị trí trong C++


Cho mảng và số k. Xoay mảng sang trái k vị trí.

Ví dụ: [5,4,7,8,5,6,8,7,9,2], k=2 → [7,8,5,6,8,7,9,2,5,4].

Thuật toán

  • k = k % n.
  • Tạo mảng mới: copy từ k đến n-1, rồi 0 đến k-1.

Ví dụ triển khai (C++)

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

void print_vector(vector<int> v) {
    cout << "[";
    for (int x : v) cout << x << ", ";
    cout << "]" << endl;
}

class Solution {
public:
    vector<int> solve(vector<int>& nums, int k) {
        vector<int> ret;
        int n = nums.size();
        k %= n;
        for (int i = k; i < n; i++) ret.push_back(nums[i]);
        for (int i = 0; i < k; i++) ret.push_back(nums[i]);
        return ret;
    }
};

int main() {
    Solution ob;
    vector<int> v = {5,4,7,8,5,6,8,7,9,2};
    print_vector(ob.solve(v, 3));
    return 0;
}

Đầu ra

[8, 5, 6, 8, 7, 9, 2, 5, 4, 7, ]