[문제풀이] 백준 2331 반복수열

import java.util.HashMap; import java.util.Map; import java.util.Scanner; public class 반복수열_2331 { public static void main(String args[]){ Scanner sc = new Scanner(System.in); Map<Integer,Integer> map = new HashMap<Integer,Integer>(); int A=sc.nextInt(); int P=sc.nextInt(); map.put(A, 0); int cnt=1; int tmp=A; /*HashMap에 (구한 D값, 몇번 째 들어온 수인지)로 입력 * D를 구하고 만약 기존에 맵에 있던 값이면 그 값의 data를 출력 후 리턴 */ while(true){ int D=0; while(tmp!=0){ //다음 D 구하기 D+=Math.pow(tmp%10,P); tmp/=10; } System.out.println(D); if(map.containsKey(D)){ System.out.println(map.get(D)); return; } map.put(D, cnt++); tmp=D; } } }