Programmers [Java] - Lv. 0 : 콜라츠 수열 만들기

 

 

 

import java.util.ArrayList;

class Solution
{
    public int[] solution(int n)
	{
		int[] result;
		ArrayList<Integer> arrList = new ArrayList<Integer>();
		arrList.add(n);
		
		while(true)
		{
			if (n % 2 == 0)
			{
				n = n / 2;
				arrList.add(n);
			}
			
			if (n == 1)
				break;
			
			else if (n % 2 != 0)
			{
				n = (3 * n) + 1;
				arrList.add(n);
			}
		}

		result = new int[arrList.size()];

		for (int i = 0; i < result.length; i++)
		{
			result[i] = arrList.get(i).intValue();
		}

		return result;

	}
}