在 C++ 中的陣列中移動元素

Jinku Hu 2023年1月30日 2021年1月22日
  1. 使用 std::rotate 演算法在 C++ 中移動陣列中的元素
  2. 使用自定義封裝函式 std::rotate 在 C++ 中移動陣列中的元素
  3. 使用 std::rotate_copy 演算法在 C++ 中移動陣列中的元素
在 C++ 中的陣列中移動元素

本文將介紹幾種在 C++ 中如何移動陣列中的元素的方法。

使用 std::rotate 演算法在 C++ 中移動陣列中的元素

std::rotate 函式是 C++ 演算法庫的一部分,可以使用 <algorithm> 標頭檔案匯入。這個演算法將陣列元素旋轉到左側。它需要三個迭代器型別的引數,其中第二個引數指定需要作為新構造的範圍的第一個元素。同時,第一個和第三個元素是源範圍的起始和結束位置的指定符。

注意,std::rotate 可以利用 rbegin/rend 迭代器將元素移到右邊。在下面的例子中,函式在具有 10 個整數的 std::vector 物件上呼叫,並演示了兩個方向的操作。

#include <iostream>
#include <vector>
#include <array>
#include <algorithm>

using std::cout; using std::vector;
using std::array; using std::rotate;
using std::endl;

template<typename T>
void printElements(T &v)
{
    cout << "[ ";
    for (const auto &item : v) {
        cout << item << ", ";
    }
    cout << "\b\b ]" << endl;
}

int main()
{
    vector<int> vec = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10};

    printElements(vec);
    rotate(vec.begin(), vec.begin() + 3, vec.end());

    printElements(vec);
    rotate(vec.rbegin(), vec.rbegin() + 3, vec.rend());

    exit(EXIT_SUCCESS);
}

輸出:

[ 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 ]
[ 4, 5, 6, 7, 8, 9, 10, 1, 2, 3 ]

使用自定義封裝函式 std::rotate 在 C++ 中移動陣列中的元素

或者,我們可以實現封裝函式來封裝 std::rotate 演算法,並要求使用者只傳遞兩個引數-要旋轉的陣列物件和代表要移動的位置數的整數。我們也可以將傳遞的整數的符號表示為旋轉操作的方向。

在這個自定義函式中,我們任意選擇正數表示右旋,負數表示左旋。

需要注意的是,這個 rotateArrayElements 函式模板既可以在固定陣列物件上工作,也可以在用 C++ 標準庫容器構建的動態陣列物件上工作。

#include <iostream>
#include <vector>
#include <array>
#include <algorithm>

using std::cout; using std::vector;
using std::array; using std::rotate;
using std::endl;

template<typename T>
void printElements(T &v)
{
    cout << "[ ";
    for (const auto &item : v) {
        cout << item << ", ";
    }
    cout << "\b\b ]" << endl;
}

template<typename T>
int rotateArrayElements(T &v, int dir)
{
    if (dir > 0) {
        rotate(v.rbegin(), v.rbegin() + dir, v.rend());
        return 0;
    } else if (dir < 0) {
        rotate(v.begin(), v.begin() + abs(dir), v.end());
        return 0;
    } else {
        return 1;
    }
}

int main()
{
    array<int, 10> arr = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10};
    vector<int> vec = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10};

    rotateArrayElements(arr, 3);
    printElements(arr);

    rotateArrayElements(vec, -3);
    printElements(vec);


    exit(EXIT_SUCCESS);
}

輸出:

[ 8, 9, 10, 1, 2, 3, 4, 5, 6, 7 ]
[ 4, 5, 6, 7, 8, 9, 10, 1, 2, 3 ]

使用 std::rotate_copy 演算法在 C++ 中移動陣列中的元素

std::rotate_copy 演算法實現了與 std::rotate 相同的操作,只是前者將旋轉後的陣列元素複製到另一個用附加函式引數指定的範圍。

首先,我們需要宣告新的範圍,在這種情況下,選擇 std::vector 型別,建構函式取源 vector 的大小。

然後我們可以呼叫 rotate_copy 函式,其引數與我們為 std::rotate 指定的引數相同,第四個迭代器表示目標 vector 的開始。

請注意,下面的例子只演示了陣列元素的左旋。

#include <iostream>
#include <vector>
#include <array>
#include <algorithm>

using std::cout; using std::vector;
using std::array; using std::rotate;
using std::endl; using std::rotate_copy;

template<typename T>
void printElements(T &v)
{
    cout << "[ ";
    for (const auto &item : v) {
        cout << item << ", ";
    }
    cout << "\b\b ]" << endl;
}

int main()
{
    vector<int> vec1 = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10};

    printElements(vec1);
    vector<int> vec2(vec1.size());
    rotate_copy(vec1.begin(), vec1.begin() + 3, vec1.end(), vec2.begin());
    printElements(vec2);

    exit(EXIT_SUCCESS);
}
[ 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 ]
[ 4, 5, 6, 7, 8, 9, 10, 1, 2, 3 ]
Author: Jinku Hu
Jinku Hu avatar Jinku Hu avatar

Founder of DelftStack.com. Jinku has worked in the robotics and automotive industries for over 8 years. He sharpened his coding skills when he needed to do the automatic testing, data collection from remote servers and report creation from the endurance test. He is from an electrical/electronics engineering background but has expanded his interest to embedded electronics, embedded programming and front-/back-end programming.

LinkedIn

相關文章 - C++ Array