2331번: 반복수열
첫째 줄에 반복되는 부분을 제외했을 때, 수열에 남게 되는 수들의 개수를 출력한다.
www.acmicpc.net
D[1] = A
D[n] = D[n - 1]의 각 자리의 숫자를 M번 곱한 수(각 자리마다 ^M)들의 합입니다.
전 재귀를 돌려가면서 현재 숫자와 수열의 갯수를 유지하였습니다.
일의자리부터 pow(tmp, M)한 값을 next에 차례대로 더해줍니다.
next가 set에 있는 값이면 cyclestart에 next를 넣어주고 리턴합니다.
(예외처리로 D[n] = D[n - 1] 즉, 자신 다음이 자신일 경우가 있어 처리해주었습니다.)
재귀를 빠져나오면서 num == cyclestart이면 ans에 cnt - 1을 넣어주어 출력하였습니다.
[C++]
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
#include <iostream>
#include <set>
#include <cmath>
using namespace std;
set<int> s;
int N, M, cyclestart, ans;
void func(int num, int cnt) {
s.insert(num);
int n = num;
int next = 0;
while (1) {
int tmp = n % 10;
tmp = pow(tmp, M);
next += tmp;
n /= 10;
if (!n) break;
}
if (s.find(next) != s.end()) {
cyclestart = next;
if (cyclestart == num) ans = cnt - 1;
return;
}
func(next, cnt + 1);
if (num == cyclestart)
ans = cnt - 1;
}
int main() {
cin.tie(NULL); cout.tie(NULL);
ios::sync_with_stdio(false);
cin >> N >> M;
func(N, 1);
cout << ans << '\n';
return 0;
}
|
cs |
[Java]
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.HashSet;
import java.util.Set;
import java.util.StringTokenizer;
public class Main {
static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
static StringTokenizer st;
static Set<Integer> s = new HashSet<>();
static int N, M, cyclestart, ans;
static void func(int num, int cnt) {
s.add(num);
int next = 0;
int n = num;
while (true) {
next += Math.pow(n % 10, M);
n /= 10;
if (n == 0)
break;
}
if (s.contains(next)) {
if (next == num)
ans = cnt - 1;
cyclestart = next;
return;
}
func(next, cnt + 1);
if (cyclestart == num)
ans = cnt - 1;
}
static void input() throws Exception {
st = new StringTokenizer(br.readLine());
N = Integer.parseInt(st.nextToken());
M = Integer.parseInt(st.nextToken());
}
public static void main(String[] args) throws Exception {
input();
func(N, 1);
System.out.println(ans);
}
}
|
cs |
'algorithm > Implementation' 카테고리의 다른 글
| boj 10163 색종이 (0) | 2021.02.24 |
|---|---|
| boj 20055 컨베이어 벨트 위의 로봇 (0) | 2021.02.23 |
| boj 10157 자리배정 (0) | 2021.02.17 |
| boj 16918 봄버맨 (0) | 2021.02.17 |
| boj 16935 배열 돌리기 3 (0) | 2021.02.10 |