9-dhlee777

[dhlee777]

🔗 문제 링크

AC

✔️ 소요된 시간

4시간30분

문자열처리

이 문제에서 배열의 원소들이 [1,2,3,5] 이런 형식으로 주어지는데 일단이 배열을 string 객체를 통해 입력받는다 그리고 나서
이 문자열에서 숫자들을 분리해 내기 위해서 for루프문을 string 사이즈 만큼 돌리며 확인해준다. 이때 숫자가 10의자리 이상 일 수 있기때문에 그 수들을 저장해놓을 string객체를 하나 더 생성해준다.
1."[","]"의 경우 건너뛴다.
2.","가 나오기 전까지 문자들을 새로 생성한 string 객체에 계속합쳐준다.
3. ","가 나올경우 그전까지 합쳐진 string 문자열들을 stoi()를 통해 정수형으로 변환하고 벡터에 push해준다.
4. 마지막 문자는 , 표시가 없으므로 제일 마지막에 따로 벡터에 push해준다.

첫번째풀이(시간초과)

#include<iostream>
#include<vector>
#include<string>
#include<algorithm>
using namespace std;
vector<int>v;
string a,b,nm;
int main(void) {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	int t,num;
	bool flag = true;
	cin >> t;
	while (t--) {
		flag = true;
		cin >> a>>num>>b;
		for (int i = 0; i <b.size(); i++) {
			if (b[i] == '[' || b[i] == ']')
				continue;
			else if (b[i] == ',')
			{
				v.push_back(stoi(nm));
				nm.clear();
			}
			else 
				nm = nm + b[i];
				
		}
		if(!nm.empty()) v.push_back(stoi(nm));
		nm.clear();

		for (int i = 0; i < a.size(); i++) {
			if (a[i] == 'R')
				reverse(v.begin(), v.end());
			else if (a[i] = 'D')
				if (v.empty())
				{
					cout << "error"<<"\n";
					flag = false;
					break;
				}
				else v.erase(v.begin());

		}
		if (!flag) continue;
		cout << "[";
		for (int j = 0; j < v.size(); j++) {
			if (j == v.size() - 1) cout << v[j];
			else cout << v[j] << ",";
		}
		cout << "]";
		cout << "\n";
		vector<int>().swap(v);

	}
}

첫번째풀이에서 R함수 로직을 구현하기 위해 stl의 reverse함수를 쓰고 D함수를 구현하기위해 erase 함수를 써버렸더니 시간초과가 떴다. 당연한것이 원소의 개수가 십만개가 되는데 이러한 reverse ,erase 함수를 써버리면 쓸때마다 모든 원소들이 자리이동을 하게되므로 엄청난 시간복잡도가 발생할것이다.

두번째풀이

예외처리 해줄게 너무많고 문자열 처리에 시간이 엄청 오래걸렸다.R은 배열에 있는 수들의 순서를 역으로 뒤집는 함수 D는 첫번째수를 삭제하는 함수이다. 주의할점은 배열에 비어있는데 D를 할경우 error를 출력해야한다.또한 출력으로 빈배열이 나올경우 []형식으로 출력을해줘야한다.
그래서 힌트를 참고하여 풀어낸 두번째 풀이는 벡터를 가리키는 두개의 인덱스를 활용한 풀이이다. start_idx는 벡터의 첫번째원소, end_idx 는 벡터의 마지막 원소의 인덱스를 가지게 한다. 그리고 난후
1)R연산이 발생할경우
원래는 start_idx와 end_idx의 값을 서로 바꿔야하는데 시간을 줄이기위해 일단 r_cnt를 통해 R의 개수를 저장해준다.
2) D연산이 발생할경우에는
만약 빈배열인경우 삭제할게 없으므로 에러를 출력하고 다음 테스트케이스로 넘어간다.
만약 RRRD처럼 R연산이 홀수번 발생하였을경우 R연산(start_idx와 end_idx 위치바꾸기)를 한번 수행하고 넘어간다.(R연산이
짝수번 발생한경우는 원상태로 회귀하기때문에 원본상태 그대로 다음단계로 넘어간다)
if(start_idx>end_idx)의 경우 시작인덱스가 더 뒤에있으므로 start_idx--를 하여 시작인덱스 위치 이동
else if(start_idx<end_idx)의 경우 일반적인경우로 시작인덱스가 앞에있으므로 stat_idx++를 하여 시작인덱스 위치 이동
else(start_idx==end_idx) 만약 두인덱스가 같은경우에는 벡터의 원소가 하나인 상태이므로 D연산을 해줄경우 빈벡터가 되어버린다 이를 표현해주기 위해 end_idx=-1로 설정해준다.

여기서 시간초과를 줄일방법은 RRRRD 이런식으로 R이 짝수개가 나와버리면 결국 원상태로 회귀하므로 R연산을 실행할 필요가 없다 즉 R의개수가 홀수개 나올때만 R연산을 한번 시행해주면 된다.

📚 새롭게 알게된 내용

처음에 풀때 stl 내장함수를 사용하여 풀었지만 시간초과만 계속났다. 인덱스를 활용하여 이렇게 배열원소를 이동시키지 않고 배열의 원소가 이동한것 처럼 보이는게 되게 신기한 풀이방법 같았다.

[InSange]

맞습니다. 해당 배열을 뒤집고하면 최악의 경우 O(N)이 N번 발생하게 되고 t도 넣게 되면 O(N^3)이 발생하게 됩니다.

투 포인터로 처음과 뒤에 인덱스를 먼저 선정한뒤 뒤집었는지 판별 여부만 확인하여 인덱스를 줄여준다면 쉽게 해결할 수 있습니다. 해당 부분을 잘 집은 것 같네요
아래는 제 풀이 입니다.

#include <iostream>
#include <vector>
#include <string>

using namespace std;

int T, n, front, last, i, j;
string p, x, num;
vector<int> v;
bool removeFront, pass;

void Solve()
{
	cin >> T;

	while (T--)
	{
		v.clear();
		num = "";
		removeFront = true;
		
		cin >> p >> n >> x;
		
		for (i = 0; i < x.length(); i++)
		{
			if (x[i] >= '0' && x[i] <= '9')
			{
				num += x[i];
			}
			else if (x[i] == ',' || x[i] == ']' && num != "")
			{
				v.push_back(stoi(num));
				num = "";
			}
		}

		pass = true;
		front = 0, last = v.size();

		for (i = 0; i < p.length(); i++)
		{
			if (p[i] == 'R')
			{
				removeFront = (removeFront ? false : true);
			}
			else
			{
				if (front == last)
				{
					cout << "error" << "\n";
					pass = false;
					break;
				}
				if (removeFront) front++;
				else last--;
			}
		}

		if (pass)
		{
			vector<int> ans;
			int len = last - front;

			cout << "[";
			for (i = 0; i < len; i++)
			{
				cout << (removeFront ? v[front++] : v[--last]);				
				if (i != len-1) cout << ",";
			}
			cout << "]\n";
		}
	}
}

int main()
{
	cin.tie(nullptr);
	ios::sync_with_stdio(false);

	Solve();

	return 0;
}

[seongwon030]

예외처리 하는 부분 인상깊었습니다.
저는 이렇게 한 번 풀어봤습니다.

import sys
from collections import deque

input = sys.stdin.readline

T = int(input())
for i in range(T):
  hi = True
  er = 'y'
  a = list(input().strip())
  n = int(input())
  s = input().strip()[1:-1]
  if not s:
    if 'D' in a:
      er = 'error'
    else:
      er = 'y'
  else:
    s = deque((map(int,s.split(','))))
    for j in range(len(a)):
      if a[j]=='R':
        if hi:
          hi = False
        else:
          hi = True
      else:
        if not s:
          er = 'error'
        else:
          if hi:
            s.popleft()
          else:
            s.pop()

    if hi == False:
      s.reverse()
    else:
      s = s
  if er=='error':
    print(er)
  else:
    s = list(s)
    re = str(s).replace(" ", "")
    print(re)

[yuyu0830]

이런 구현 문제는 항상 전체적인 로직을 먼저 구상하고, 세세한 부분을 하나씩 구현하는 Top-down 방식으로 구현해야하는 것 같아요.. 문제 분할을 잘 하고 분할된 문제를 해결하는 방법을 잘 세우는게 중요한 것 같습니다
이번 문제도 Reverse를 어떻게 잘 구현하느냐가 메인 문제였던 것 같아요! 실제 요소들을 다 바꿀 필요 없이 포인터를 잘 사용해서 데이터는 그대로 두고 원하는 방식대로 접근하는 방식은 자주 사용되고, 다양하게 응용할 수 있으니 연습 많이 해보시는걸 추천드립니다 :)