동일한 'a' 및 'b' 개수로 업데이트된 문자열을 가져오는 C++ 코드

길이가 n인 문자열 S가 있다고 가정합니다. S에는 두 가지 유형의 문자 'a'와 'b'만 포함됩니다. 길이의 모든 접두사가 동일한 양의 문자 'a'와 'b'를 갖도록 문자열을 수정하려고 합니다. 이를 달성하기 위해 다음 작업을 임의의 횟수만큼 수행할 수 있습니다. 문자열의 일부 위치를 선택하고 이 위치의 문자를 다른 문자로 교체합니다. 업데이트된 문자열을 반환합니다.

따라서 입력이 S ="aabbbb"와 같으면 출력은 "baabab"이 됩니다.

단계

이 문제를 해결하기 위해 다음 단계를 따릅니다. −

n := size of S
for initialize i := 0, when i < n, update i := i + 2, do:
   if S[i] is same as S[i + 1], then:
      (increase ans by 1)
   S[i] := (if S[i] is same as 'a', then 'b', otherwise 'a')
return S

예시

이해를 돕기 위해 다음 구현을 살펴보겠습니다. −

#include <bits/stdc++.h>
using namespace std;
string solve(string S){
   int n = S.size(), ans = 0;
   for (int i = 0; i < n; i += 2)
      if (S[i] == S[i + 1]){
         ans++;
         S[i] = S[i] == 'a' ? 'b' : 'a';
      }
   return S;
}
int main(){
   string S = "aabbbb";
   cout << solve(S) << endl;
}

입력

"aabbbb"

출력

baabab