PS/DP
[11051] (C++) 이항 계수2
프레딕
2023. 2. 14. 23:25
728x90
https://www.acmicpc.net/problem/11051
문제
자연수 N과 정수 K가 주어졌을 때 이항 계수 (NK)를 10,007로 나눈 나머지를 구하는 프로그램을 작성하시오.
입력
첫째 줄에 N과 K가 주어진다. (1 ≤ N ≤ 1,000, 0 ≤ K ≤ N)
출력
(NK)를 10,007로 나눈 나머지를 출력한다.
코드
#include <algorithm>
#include <cmath>
#include <iostream>
#include <map>
#include <set>
#include <vector>
using namespace std;
int N;
int K;
long long DP[1001][1001];
int main(void) {
cin >> N >> K;
for(int i=0; i<=N; ++i){
for(int j=0; j<=i; ++j){
if(j == 0) DP[i][0] = 1;
else if(j == i) DP[i][j] = 1;
else{
DP[i][j] = (DP[i-1][j-1] + DP[i-1][j]) % 10007;
}
}
}
cout << DP[N][K];
return 0;
}
이 문제는 DP를 활용한 문제이다. 동적 계획법을 쓴 대표적 문제이다
728x90
반응형