- 문제
선영이는 주말에 할 일이 없어서 새로운 언어 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를 통해 간단하게 문자열을 정수로 바꿀 수 있다.
#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;
}
뭔소린지 잘 모르겠다.... 복습할 때 분석해보기
좋은 정보 얻어갑니다, 감사합니다.