컴퓨터 공학/C++

C++에서 동적계획법을 이용한 이항계수 구하는 코드

혼새미로 2015. 11. 27. 01:40

 

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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
//이항계수 n개중 k 개 선택하는 방법
#include <iostream>
using namespace std;
 
#define N 20
#define K 5
 
 
int bin(int n,int k)
{
    if(k==0 || n==k)
        return 1;
    else
        return bin(n-1,k-1)+bin(n-1,k);
}
 
int dBin(int i,int j,int** arr)
{
    if(i==j || j==0)
    {
        return 1;
    }
 
    return arr[i-1][j-1]+arr[i-1][j];
 
}
 
int main()
{
    int** arr=new int*[N+1];
    for(int i=0;i<=N;i++)
    {
        arr[i]=new int[K+1];
    }
    for(int i=0;i<=N;i++)
    {
        for(int j=0;j<=K;j++)
        {
            arr[i][j]=0;
        }
    }
    arr[0][0]=1;
    arr[1][0]=1;
 
    for(int i=1;i<=N;i++)
    {
        for(int j=0;j<=i&&j<=K;j++)
        {
            arr[i][j]=dBin(i,j,arr);
            cout<<"arr["<<i<<"]["<<j<<"]="<<arr[i][j]<<"\n";
        }
    }
 
    cout<<"result:"<<arr[N][K]<<"\n";
 
    for(int i=0;i<=N;i++)
    {
        delete[] arr[i];
    }
    delete[] arr;
 
    return 0;
}