1-5-字符串的旋转

/*
    题目描述：
        对于一个字符串，和字符串中的某一位置，请设计一个算法，将包括i位置在内的左侧部分移动到右边，
        将右侧部分移动到左边。给定字符串A和它的长度n以及特定位置p，请返回旋转后的结果。
    测试样例：
        "ABCDEFGH",8,4
        返回："FGHABCDE"
*/
/*
    eg: A = "ABCDEFGH", n = 8, p = 4
    首先把A反转， A = "HGFEDCBA";
    然后把[0,n-1-p-1] 和 [n-1-p, n-1]分别反战：
    得到 FGHABCDE。
    时间复杂度为O(n),空间复杂度为O(1)。
    使用reverse函数反转。
*/
#include <iostream>
#include <string>
using namespace std;

string rotateString(string A, int n, int p) {
    if (n <= 0 || p <= 0 || p >= n)
        return "";
    reverse(A.begin(), A.end());
    reverse(A.begin(), A.begin()+n-1-p);
    reverse(A.begin()+n-1-p, A.end());
    return A;
}

int main(){
    string a;
    cin >> a;
    cout << rotateString(a,a.length(), 4);
    return 0;
}