취미로 코딩하기

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

이번에 볼 문제는 백준 11560번 문제인 다항식 게임이다.
문제는 아래 링크를 확인하자.

www.acmicpc.net/problem/11560

이 문제는 다항식의 전개식에서 계수를 찾을 것을 요구하고 있다.

수의 범위가 충분히 작으므로(곱해지는 다항식의 수가 최대 20개, 나올 수 있는 최대 차수가 210차항), knapsack 문제와 같은 방식의 풀이를 적용해서 문제를 해결할 수 있다.

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

#include <iostream>
#include <cstring>
using namespace std;

long long arr[231];

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

	int T; cin >> T;
	while (T--) {
		memset(arr, 0, sizeof(arr));
		arr[0] = arr[1] = 1;

		int K, N; cin >> K >> N;
		for (int k = 2; k <= K; k++) {
			for (int i = 210; i >= 0; i--) {
				for (int j = 1; j <= k; j++) {
					arr[i + j] += arr[i];
				}
			}
		}
		cout << arr[N] << '\n';
	}
}