취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 11583번 문제인 인경호의 징검다리이다.
문제는 아래 링크를 확인하자.

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

어떤 양의 정수의 trailing zero의 개수는 주어진 수가 10을 얼마나 많이 소인수로 가지고 있는지와 같다. 이는 그 수의 2와 5 각각의 개수의 최솟값과 같다.

따라서 문제에서 주어진 과정으로 얻을 수 있는 수 가운데 trailing zero의 개수가 최소인 수 중에는 2의 개수가 가장 적게 들어있거나 5의 개수가 가장 적게 들어있는 수가 있음을 관찰할 수 있다.

따라서 가능한 2를 가장 적게 포함하는 수와 5를 가장 적게 포함하는 수의 2와 5의 개수를 각각 구해 문제를 해결할 수 있다.

아래는 제출한 소스코드이다.

#include <iostream>
using namespace std;

int N, K;
int A[200001], B[200001];

void solve() {
	cin >> N >> K;
	for (int i = 0; i < N; i++) {
		int x; cin >> x;
		int cnt2 = 0, cnt5 = 0;
		while (x % 2 == 0) {
			x /= 2, cnt2++;
		}
		A[i] = 0x3f3f3f3f;
		for (int k = max(0, i - K); k < i; k++) {
			A[i] = min(A[i], A[k]);
		}
		if (A[i] == 0x3f3f3f3f) A[i] = 0;
		A[i] += cnt2;
		while (x % 5 == 0) {
			x /= 5, cnt5++;
		}
		B[i] = 0x3f3f3f3f;
		for (int k = max(0, i - K); k < i; k++) {
			B[i] = min(B[i], B[k]);
		}
		if (B[i] == 0x3f3f3f3f) B[i] = 0;
		B[i] += cnt5;
	}
	cout << min(A[N - 1], B[N - 1]) << '\n';
}

int T;

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);

	cin >> T;
	while (T--) solve();
}