[백준 / BOJ] C++ 6500 랜덤 숫자 만들기

6500번: 랜덤 숫자 만들기

 

문제

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

6500번: 랜덤 숫자 만들기

 

 

풀이

a0을 입력받고 a0 * a0을 앞에 0을 추가해 8자리 숫자로 바꿔준 다음 가운데 4자리를 a1으로 한다. 이를 반복해 가며 가능한 ai의 수를 구하는 문제.

 

숫자 N을 입력받고 N*N의 가운데 4자리를 다시 N으로 하며 모든 N을 계속 배열에 넣었다. 그러다 N이 이미 배열 안에 있을 경우 반복문을 탈출한다. 배열의 크기를 출력하면 된다.

 

 

코드

#include <bits/stdc++.h>
using namespace std;

int main() {
  ios::sync_with_stdio(false);
  cin.tie(NULL);
  while (1) {
    int n;
    cin >> n;
    if (!n)
      break;
    vector<int> v;
    v.push_back(n);
    n *= n;
    while (1) {
      string s = to_string(n);
      while (s.size() < 8)
        s = "0" + s;
      n = stoi(s.substr(2, 4));
      if (find(v.begin(), v.end(), n) != v.end())
        break;
      v.push_back(n);
      n *= n;
    }
    cout << v.size() << '\n';
  }
  return 0;
}