[SW Expert Academy] - (D3)4676. 늘어지는 소리 만들기

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AWRKWITqfvIDFAV8&categoryId=AWRKWITqfvIDFAV8&categoryType=CODE

SW Expert Academy

코드

import java.util.Arrays;
import java.util.Scanner;

public class Solution {

	public static void main(String[] args) {
		Scanner scan = new Scanner(System.in);
		
		int t = scan.nextInt();
		
		for(int tc=1; tc<=t; tc++) {
			String str = scan.next();
			StringBuilder sb = new StringBuilder(str);
//			StringBuilder sb2 = new StringBuilder(str);
			int H = scan.nextInt();	// 삽입할 하이픈의 갯수
			int[] arr = new int[H]; // 하이픈을 넣을 위치
			
			for(int i=0; i<H; i++) {
				arr[i] = scan.nextInt();
			}
			
			Arrays.sort(arr);
			
			for(int i=0; i<H; i++) {
				sb.insert(arr[i] + i, "-");
//				System.out.println(sb.toString());
//				str = sb.toString();
//				sb = sb2;
			}
			
			System.out.println("#" + tc + " " + sb.toString());
		}
		
		scan.close();
	}

}

풀이

문제는 문자열과 삽입할 하이픈("-")의 갯수가 주어지고, 하이픈의 갯수만큼 하이픈을 넣을 위치가 주어진다.

각각의 위치에 맞게 하이픈을 삽입해주면 된다.

 

우선 문제 풀이는 다음과 같다.

String 클래스에는 특정 위치에 데이터를 삽입하는 메소드가 없어서 StringBuilder 클래스를 활용했다.

StringBuilder 클래스의 insert() 메소드는 다양한 매개변수로 오버로딩이 되어있는데,

대표적으로 두개의 매개변수중 첫번째 매개변수에는 삽입할 위치, 두번째 위치에는 삽입할 데이터를 뜻한다.

즉, StringBuilder.insert(1, "A"); 라는 메소드는 첫번째 위치에 "A"값을 삽입하라는 뜻이다.

 

하지만 문자열을 삽입하면 그만큼 문자열의 길이도 늘어나기에, 정상적으로 작동하지 않는다.

만약 문제의 첫번째 입력처럼 wow, 2 3 2 로 데이터를 삽입하면 아래와 같이 출력이된다.

삽입될때마다 문자열의 길이도 늘어나기에 이부분도 생각을 해주어야한다.

 

어떻게할지 생각해보다가 StringBuilder 클래스를 하나 더 만들어놓고 하나씩 삽입해서 저장하는 식으로 생각을했었다.

				sb.insert(arr[i] + i, "-");
				str = sb.toString();
				sb = sb2;

위 코드처럼 StringBuilder 클래스인 sb에 하이픈을 삽입하고, 문자열에 저장한 후 초기에 입력받은 문자열로 sb값을 설정하는거였는데, 이러면 하이픈이 추가가 되는것이 아닌 일시적인 값의 변화만 있었다.

 

생각보다 해결은 간단했다! 문자열에 하이픈을 삽입할때마다, 그 문자열의 길이가 증가하므로 똑같이 삽입할 위치를 늘려주면 되는 것이다!

 

예를들어, wow라는 문자열에 2 3 2 위치에 하이픈을 삽입한다면 ..

wo-w , wo-w- , wo--w- 이러한 방식으로 하이픈이 추가된다.

삽입할 위치를 늘려주는건 for문의 변수인 i값을 더해주면 된다.

i는 0부터 시작하므로 처음에는 0을, 그 후로는 하이픈을 넣을때마다 i값을 더해주면 문자열이 증가하는것과 동일하게 하이픈을 넣을 위치도 증가하게된다.

 

마지막으로 2 3 2 라는 하이픈을 삽입할 위치를 정렬을 해야한다.

만약, wow에 2 3 2라는 하이픈을 위의 과정으로 삽입을 한다면... sb.insert(arr[i] + i, "-");

i=0 ... sb.insert(2, "-"); ---> wo-w

i=1 ... sb.insert(4, "-"); ---> wo-w-

i=2 ... sb.insert(4, "-"); ---> wo-w-- 

정답은 wo--w- 이지만, wo-w-- 로 오류가 난다.

 

따라서 정렬을 해주고, 값이 작은 순서대로 하이픈을 넣어주면 된다 !