[백준] 17413 단어 뒤집기2

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

17413번: 단어 뒤집기 2

 

문제를 다 풀고나서 알고리즘 분류에 스택이 있는 걸 알았다. 사실 나는 케이스를 전체적으로 하나하나 다 나눠서 구현했지만 생각해보니 스택을 이용하면 더 효율적으로 해결할 수 있을 것 같다. 문제를 보고 최대한 자료구조를 이용하는 것이 빠르면서도 정확할 수 있기 때문에 최대한 연상하는 연습을 해야될 것 같다.

---

문제 및 입출력

---

예제1

 

---

예제2

---

​코드

import java.util.Scanner;

public class Main
{
     public static void main(String[] args)
     {
         Scanner sc = new Scanner(System.in);
         String input = sc.nextLine();
         StringBuilder sb = new StringBuilder();
         String result = "";

         for(int i=0; i<input.length(); i++)
         {
              char cmp = input.charAt(i);

              // 1. '>'를 만났을 때
              if(cmp=='>')
              {
                   if(sb.toString().contains("<"))
                   {
                        sb.append(cmp);
                        result += sb.toString();
                        sb.delete(0, sb.length());
                   }
             }
             // 2. '<'를 만났을 때
             else if(cmp=='<')
             {
                  // 2-1. 이미 '<'를 임시변수에 가지고 있을 때는 이전 값들을 거꾸로 결과값에 옮기기
                  if(sb.length()!=0)
                  {
                       result += sb.reverse().toString();
                       sb.delete(0, sb.length());
                       sb.append("<");
                  }
                  // 2-2. 그 외에는 그대로 결과값에 옮기기
                  else
                  {
                       sb.append(cmp);
                  }
             }
             // 3. ' '를 만났을 때
             else if(cmp==' ')
             {
                  if(sb.toString().contains("<"))
                  {
                       sb.append(cmp);
                  }
                  else
                  {
                       result += sb.reverse().toString() + " ";
                       sb.delete(0, sb.length());
                  }
             }
             // 4. 맨 마지막 인덱스인데 아직 값이 있을 때
             else if(i==input.length()-1 && sb.length()!=0) 
             {
                  sb.append(cmp);
                  result += sb.reverse().toString();
             }
             // 5.  그 외
             else
             {
                  sb.append(cmp);
             }
        }

        System.out.println(result);
        sc.close();
   }
}