1-5-字符串的旋轉

 1 /*
 2     題目描述:
 3         對於一個字符串,和字符串中的某一位置,請設計一個算法,將包括i位置在內的左側部分移動到右邊,
 4         將右側部分移動到左邊。給定字符串A和它的長度n以及特定位置p,請返回旋轉後的結果。
 5     測試樣例:
 6         "ABCDEFGH",8,4
 7         返回:"FGHABCDE"
 8 */
 9 /*
10     eg: A = "ABCDEFGH", n = 8, p = 4
11     首先把A反轉, A = "HGFEDCBA";
12     而後把[0,n-1-p-1] 和 [n-1-p, n-1]分別反戰:
13     獲得 FGHABCDE。
14     時間複雜度爲O(n),空間複雜度爲O(1)。
15     使用reverse函數反轉。
16 */
17 #include <iostream>
18 #include <string>
19 using namespace std;
20 
21 string rotateString(string A, int n, int p) {
22     if (n <= 0 || p <= 0 || p >= n)
23         return "";
24     reverse(A.begin(), A.end());
25     reverse(A.begin(), A.begin()+n-1-p);
26     reverse(A.begin()+n-1-p, A.end());
27     return A;
28 }
29 
30 int main(){
31     string a;
32     cin >> a;
33     cout << rotateString(a,a.length(), 4);
34     return 0;
35 }
相關文章
相關標籤/搜索