취미로 코딩하기

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

이번에 볼 문제는 백준 2410번 문제인 2의 멱수의 합이다.
문제는 아래 링크를 확인하자.

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

1부터 100만까지, '식이 1로 끝날 경우의 수', '식이 1 또는 2로 끝날 경우의 수', '식이 1 또는 2 또는 4로 끝날 경우의 수' ... 를 순서대로 DP를 이용하여 계산하는 것으로 문제를 해결할 수 있다.

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

#include <iostream>
using namespace std;

int arr[1000001];

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

	arr[0] = 1;
	
	int k = 1;
	while (k < 1000000) {
		for (int i = k; i <= 1000000; i++) {
			arr[i] = (arr[i] + arr[i - k]) % 1000000000;
		}
		k <<= 1;
	}

	int N; cin >> N;
	cout << arr[N];
}