다음 코딩테스트 문제풀이

// 다음 개발자 코딩 테스트 2011년 2 영역 2번 문제를 풀어보았다.
import java.util.ArrayList;

import java.util.Scanner;





class Test {

	static int in;

	public static void main(String[] args) {

		Scanner scanner = new Scanner(System.in);

		String str = scanner.nextLine();

		in = Integer.parseInt(str);

		System.out.println(B(in).toString());

		

	}

	

	static int A(String str){

		int num = 0;

		int sum = 0;

		for (int i = 0; i < str.length(); i++) {

			num = Integer.parseInt(str.substring(i,i+1));

			num*=num;

			sum+=num;

		}

		return sum;

		

	}

	

	static ArrayList
 B(int num){

		ArrayList intArr = new ArrayList<>();

		int isNum = num;

		int isNum1 = 1;

		

		while(isNum!=isNum1 || -1==intArr.indexOf(isNum)){

			isNum = A(isNum+"");

			

			if (-1==intArr.indexOf(isNum)) {

				intArr.add(isNum);

			}else{

				System.out.println("예외처리");

				break;

			}

		}

		

		return intArr;
	}
}