백준

백준 2331 반복수열(Java)

Park DJ 2023. 1. 29. 17:23

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

 

2331번: 반복수열

첫째 줄에 반복되는 부분을 제외했을 때, 수열에 남게 되는 수들의 개수를 출력한다.

www.acmicpc.net


 

해석 및 팁

 


 

이 문제는 공식에 따라 수를 구한 다음 ArrayList에 값이 존재하지 않으면 add 해주고 존재하면 원래 존재하던 값의 index를 출력해 주면 되는 문제입니다.

 


 

Java 코드

 


import java.util.Scanner;
import java.util.ArrayList;

public class Main {
  public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);
    StringBuilder sb = new StringBuilder();
    ArrayList<Integer> list = new ArrayList<>(); 

    int a = sc.nextInt();
    double p = sc.nextInt();

    list.add(a);

    while(true) {
      int sum = 0;
      int num = list.get(list.size() - 1);
      int len = Integer.toString(num).length();

      for(int i = 0; i < len; i++) {
        sum += Math.pow(num % 10, p);
        num /= 10;
      }

      if(list.contains(sum)) {
        sb.append(list.indexOf(sum));
        break;
      }

      else list.add(sum);
    }
    
    System.out.println(sb);
  }
}