[백준] 1551번: 수열의 변화(수학, 시뮬레이션)

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

1551번: 수열의 변화

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {

	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(bf.readLine());
		
		/** 1 **/
		int N = Integer.parseInt(st.nextToken());
		int K = Integer.parseInt(st.nextToken());
		st = new StringTokenizer(bf.readLine() , ",");	// ,(콤마)로 자르기.
		int[] arr = new int[N];
		for(int i=0; i<arr.length; i++)
			arr[i] = Integer.parseInt(st.nextToken());
		
		/** 2 **/
		int[] newArr = Arrays.copyOf(arr, arr.length);
		int arrLength = N-1;
		
		/** 3 **/
		while(K != 0) {
			int[] moveArr = new int[arrLength];
			
			// B[i] = A[i+1] - A[i]
			for(int i=0; i<newArr.length-1; i++) 
				moveArr[i] = newArr[i+1] - newArr[i];
			
			arrLength --;
			K --;
			newArr = Arrays.copyOf(moveArr, moveArr.length);
		}
		
		/** 4 **/
		for(int i=0; i<newArr.length; i++) {
			if(i != newArr.length-1) 
				System.out.print(newArr[i]+",");
			else
				System.out.println(newArr[i]);
		}
		bf.close();
	}

}

풀이

생각보다 꽤 까다롭게 구현했다 ..

크기가 N인 수열 A가 주어졌을때, B[i] = A[i+1] - A[i]의 규칙을 가지고, K번만큼 반복해서 시행했을때 나오는 수열을 출력하는 문제다.

따라서 문제 그대로 구현을 했는데,,

 

/** 1 **/

변수 선언 부분이다.

문제에서 입력받을때 ,를 기준으로 받으므로 StringTokenizer의 두번째 매개 변수에 ","를 추가해 콤마를 자른다.

int[] arr = 처음 입력받는 수를 저장할 배열

 

/** 2 **/

int[] newArr = 처음 입력받은 arr 배열을 복사하고,

추 후 while문에서 새로운 배열로 옮길때 사용할 배열.

int arrLength = 배열의 길이로, 한번씩 수행할때마다 감소하고, 첫 배열의 크기는 N-1 이다.

 

/** 3 **/

K가 0이될때까지 반복하는데,

moveArr 은 기존의 newArr 배열을 B[i] = A[i+1] - A[i] 의 규칙에 따라 옮겨서 저장할 배열이다.

그 후 한번 수행마다 배열의 길이가 1씩 감소하고, K를 1씩 감소시킨다.

위 과정이 끝나면 newArr 배열에 moveArr에 저장된 값들을 복사한다.

(moveArr 배열은 while문을 돌때마다 새로 만들어지므로, 비교할 배열은 newArr이 된다.)

 

/** 4 **/

출력할때도 ,(콤마)로 구분해서 출력해야 하는데, 배열의 마지막값에는 ,가 들어가면 안되므로 조건을 추가했다.

 

 

 

다른 풀이

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main{
	public static int SIZE;
	public static int[] arr;
	public static void main(String[] args) throws Exception {
		BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(in.readLine());
		int i=0, n = Integer.parseInt(st.nextToken()), k = Integer.parseInt(st.nextToken());
		SIZE = n; arr = new int[n];
		st = new StringTokenizer(in.readLine());
		while(st.hasMoreTokens()) arr[i++] = Integer.parseInt(st.nextToken(","));
		
		for(i=0;i<k;i++) transform();
		i=0;
		while(true){
			System.out.print(arr[i++]);
			if(i==SIZE) break;
			System.out.print(",");
		}
	}
	private static void transform(){
		--SIZE;
		for(int i=0;i<SIZE;i++) arr[i] = arr[i+1] - arr[i];
	}
}