백준 8958번 C++

백준 8958번 OX퀴즈

https://www.acmicpc.net/problem/8958

8958번: OX퀴즈

정답 코드

#include<iostream>
#include<cstring>
using namespace std;
int ret(char *a) {
	int result = 0, index, max1,dif;
	int x = 0;
	 while(x <strlen(a)) {
		 if (a[x] == 'O') {
			 index = x;
			 dif = x;
			 while (index < strlen(a)) {
				 index++;
				 if (index == strlen(a)) {
					 max1 = index;
					 break;
				 }
				 else if (a[index] == 'X') {
					 max1 = index;
					 break;
				 }
			 }
			 for (int k = 1; k<=max1 - dif; k++)
				 result += k;
			 x = index;
		 }
		 else
			 x++;
	}

	return result;

}
int main() {
	int num;
	cin >> num;
	for (int x = 0; x < num; x++) {
		char input[81];
		cin >> input;
		int a = ret(input);
		cout << a << endl;
	}
}

문제 풀이

문자열을 입력받고 ret함수로 넘겨줍니다.
ret함수에서는 문자열의 길이만큼 while문을 돌려주고 그 안에서 O와 X를 찾습니다. O를 찾게 되면 그 지점부터 while문을 돌려서 X가 나오거나 문자열의 길이를 벗어나게 될 때까지 index값을 ++해주고 max1에 저장해줍니다. 이 while문을 탈출하면 for문을 돌려서 result에 k를 더해줍니다. 이 과정을 계속 반복하여
문제를 해결하였습니다.