[LeetCode] 557. Reverse Words in a String III

https://leetcode.com/problems/reverse-words-in-a-string-iii/description/

Reverse Words in a String III - LeetCode

Reverse Words in a String III - Easy

 

 

주어진 문자열을 단어내의 알파벳만 순서를 뒤집어서 단어와 띄어쓰기 위치는 보존한 채, 재배열하는 문제다.

 

단어의 시작을 나타내는 b라는 변수를 선언하고,

앞에서부터 훑으며 띄어쓰기가 있을 경우

b에서 부터 띄어쓰기 직전까지 revserse해 저장하고,

b를 이 다음으로 옮겨주자.

 

class Solution {
public:
    string reverseWords(string s) {
        int b = 0;
        for (int i=0; i<s.size(); i++) {
            if (s[i] == ' ') {
                reverse(s.begin() + b, s.begin() + i);
                b = i + 1;
            }
        }
        reverse(s.begin() + b, s.end());
        return s;
    }
};