BOJ 5430 (AC)

JH·2023년 7월 25일
0

BOJ 알고리즘 (C++)

목록 보기
85/97
post-custom-banner

  • 문제
    선영이는 주말에 할 일이 없어서 새로운 언어 AC를 만들었다. AC는 정수 배열에 연산을 하기 위해 만든 언어이다. 이 언어에는 두 가지 함수 R(뒤집기)과 D(버리기)가 있다.

    함수 R은 배열에 있는 수의 순서를 뒤집는 함수이고, D는 첫 번째 수를 버리는 함수이다. 배열이 비어있는데 D를 사용한 경우에는 에러가 발생한다.

    함수는 조합해서 한 번에 사용할 수 있다. 예를 들어, "AB"는 A를 수행한 다음에 바로 이어서 B를 수행하는 함수이다. 예를 들어, "RDD"는 배열을 뒤집은 다음 처음 두 수를 버리는 함수이다.

    배열의 초기값과 수행할 함수가 주어졌을 때, 최종 결과를 구하는 프로그램을 작성하시오.

  • 입력
    첫째 줄에 테스트 케이스의 개수 T가 주어진다. T는 최대 100이다.

    각 테스트 케이스의 첫째 줄에는 수행할 함수 p가 주어진다. p의 길이는 1보다 크거나 같고,
    100,000보다 작거나 같다.

    다음 줄에는 배열에 들어있는 수의 개수 n이 주어진다. (0 ≤ n ≤ 100,000)

    다음 줄에는 [x1,...,xn]과 같은 형태로 배열에 들어있는 정수가 주어진다. (1 ≤ xi ≤ 100)

    전체 테스트 케이스에 주어지는 p의 길이의 합과 n의 합은 70만을 넘지 않는다.

  • 출력
    각 테스트 케이스에 대해서, 입력으로 주어진 정수 배열에 함수를 수행한 결과를 출력한다. 만약, 에러가 발생한 경우에는 error를 출력한다.

처음 풀이 (실패)

#include<iostream>
#include<cstring>
#include<queue>
#include<cmath>
#include<algorithm>
#include<stack>

using namespace std;
int TestCount;
string func;
deque<int> dq;
deque<int> answer;
stack<int> temp;
void fast_io() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL); cout.tie(NULL);
}

void input() {
	cin >> TestCount;
}

void print() {
	for (int i = 0; i < answer.size(); i++)
	{
		cout << answer[i];
	}
	cout << '\n';
	
}

void solve() {
	while (TestCount--)
	{
		int resourceCount;
		int num = 0; int cnt = 0;
		string resources;
		cin >> func >> resourceCount
			>> resources;
		for (int i = 1; i < resources.size(); i++)
		{
			if (resources[i] != ',')
			{
				num = num * pow(10, cnt) + (resources[i] - '0');
				cnt++;
				if (i == resources.size() - 2)
				{
					dq.push_back(num);
					break;
				}
			}
			else
			{
				cnt = 0;
				dq.push_back(num);
				num = 0;
			}
		}

		for (int i = 0; i < func.size(); i++)
		{
			if (func[i] == 'R')
			{
				while (!dq.empty())
				{
					temp.push(dq.front());
					dq.pop_front();
				}

				while (!temp.empty())
				{
					dq.push_back(temp.top());
					temp.pop();
				}
			}
			else {
				if (dq.size() == 0)
				{
					cout << "error";
					break;
				}
				dq.pop_front();
			}
		}

		
		for (int i = 0; i < dq.size(); i++)
		{
			answer.push_back(dq[i]);
		}
		print();
		dq.clear();
		answer.clear();
	}
}

int main() {
	fast_io();
	input();
	solve();

	return 0;
}

   입력에서 '[' ']' ','를 제거하기 위해 숫자로 변환하고 dq에 넣어 입력으로 주어진 function을 수행 후 다시 answer queue에 넣으려고 시도 했으나
출력에서 다시 대괄호와 콤마를 넣는 과정에서 int와 char 형식이 섞여 숫자 부분이 제대로 출력되지 않았다.

위 코드를 실행하면 콤마와 괄호 빼고는 숫자 자체는 제대로 나오는 것 같다. 그러나 함수의 길이, 배열의 길이 모두 10만이기 때문에 시간복잡도 O(NM)이 나온다면 시간초과가 뜰 것이다.
(괄호를 제대로 넣더라도 뒤집기 위해 for문과 while문이 중첩된 부분에서 아마 시간 초과가 나오지 않을까 싶다)

성공한 풀이

#include<iostream>
#include<string>
#include<queue>
#include<algorithm>


using namespace std;
int TestCount;
string func;
deque<int> dq;

bool errorFlag;
bool reverseFlag;

void fast_io() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL); cout.tie(NULL);
}

void input() {
	cin >> TestCount;
}

void print() {
	if (!errorFlag)
	{
		if (reverseFlag)
		{
			cout << '[';
			while (!dq.empty())
			{
				cout << dq.back();
				dq.pop_back();
				if (dq.empty())
				{
					break;
				}
				cout << ',';
			}
			cout << ']' << '\n';
		}
		else
		{
			cout << '[';
			while (!dq.empty())
			{
				cout << dq.front();
				dq.pop_front();
				if (dq.empty())
				{
					break;
				}
				cout << ',';
			}
			cout << ']' << '\n';
		}
	}

}

void solve() {
	while (TestCount--)
	{
		int resourceCount;
		string temp;
		string resources;
		cin >> func >> resourceCount
			>> resources;
		for (int i = 1; i < resources.size(); i++)
		{
			if (resources[i] >= '0' && resources[i] <= '9')
			{
				temp += resources[i];
			}
			else {
				if (!temp.empty())
				{
					dq.push_back(stoi(temp));
					temp.clear();
				}
			}
		}

		for (int i = 0; i < func.length(); i++)
		{
			if (func[i] == 'R')
			{
				reverseFlag = !reverseFlag;
			}
			else
			{
				if (dq.empty())
				{
					errorFlag = true;
					cout << "error" << '\n';
					break;
				}
				else
				{
					if (reverseFlag)
					{
						dq.pop_back();
					}
					else
					{
						dq.pop_front();
					}
				}
			}

		}
		print();
		dq.clear();
		errorFlag = false;
		reverseFlag = false;
	}
}

int main() {
	fast_io();
	input();
	solve();

	return 0;
}

   위에서 에러와 순서 뒤집는 부분을 flag 변수를 두어 정방향 역방향에 따라 출력을 하도록 했다. 위에서 숫자를 넣을 때 num이 3자리 이상이면 (pow,cnt)에서 자릿수가 뻥튀기 되는 문제가 생겼다. 이는 stoi를 통해 간단하게 문자열을 정수로 바꿀 수 있다.

시간 복잡도 : O(NM) -> (M : TEST 수)


숏코딩
#include <bits/stdc++.h>
using namespace std;

const int M = 100000;

int dq[M];
int d, i, j;
char c;

bool error() {
	string s; cin >> s;
	cin >> j >> c;
	for (i = 0; i < j; ++i)
		cin >> dq[i] >> c;
	if (!j) cin >> c;

	i = 0, d = 1;
	for (char& c : s) {
		if (c == 'R') {
			swap(i, j);
			j += (d = -d);
		}
		else if (i == j) return true;
		i += d;
	}

	for (cout << '['; i ^ j; i += d) {
		cout << dq[i];
		if ((i + d) ^ j) cout << ',';
	}
	cout << "]\n";
	return false;
}
int main() {
	cin.tie(0), cout.tie(0);
	ios::sync_with_stdio(0);

	int T; cin >> T;
	while (T--)
		if (error())
			cout << "error\n";
	return 0;
}

   뭔소린지 잘 모르겠다.... 복습할 때 분석해보기

profile
블로그 -> 노션
post-custom-banner

2개의 댓글

comment-user-thumbnail
2023년 7월 25일

좋은 정보 얻어갑니다, 감사합니다.

1개의 답글