Equal Sum Sets (DFS)

来源:互联网 发布:淘宝实名认证在哪里看 编辑:程序博客网 时间:2024/04/30 02:20

UvaLive 6661 Equal Sum Sets (DFS)

 

 

Let us consider sets of positive integers less than or equal to n. Note that all elements of a set are different. Also note that the order of elements doesnt matter, that is, both {3, 5, 9} and {5, 9, 3} mean the same set.

Specifying the number of set elements and their sum to be k and s, respectively, sets satisfying the conditions are limited. When n = 9, k = 3 and s = 23, {6, 8, 9} is the only such set. There may be more than one such set, in general, however. When n = 9, k = 3 and s = 22, both {5, 8, 9} and {6, 7, 9} are possible.

You have to write a program that calculates the number of the sets that satisfy the given conditions.

Input

The input consists of multiple datasets. The number of datasets does not exceed 100. Each of the datasets has three integers n, k and s in one line, separated by a space. You may assume 1 ≤ n ≤ 20, 1 ≤ k ≤ 10 and 1 ≤ s ≤ 155. The end of the input is indicated by a line containing three zeros.

Output The output for each dataset should be a line containing a single integer that gives the number of the sets that satisfy the conditions. No other characters should appear in the output. You can assume that the number of sets does not exceed 231 − 1.

Sample Input

9 3 23

9 3 22

10 3 28

16 10 107

20 8 102

20 10 105

20 10 155

3 4 3

4 2 11

0 0 0

Sample Output

1

2

0

20

1542

5448

1

0

0

 

 

题意:

求从不超过 N 的正整数当中选取 K 个不同的数字,组成和为 S 的方法数。

1 <= N <= 20  1 <= K<= 10  1 <= S <= 155

 

AC代码

复制代码
#include<iostream>using namespace std;int n,k,s,total;void dfs(int x,int y,int z){    if(y==k&&z==s)    {        total++;        return;    }    for(int i=1; i<=x-1; i++)    {        if(z+i<=s)        dfs(i,y+1,z+i);    }}int main(){    while(cin>>n>>k>>s&&n&&k&&s)    {        total=0;        dfs(n+1,0,0);        cout<<total<<endl;    }    return 0;}

由于n,k,s都不大,直接递推出所有结果,递推的方法类似于背包问题。

dp[i][j][k]表示元素和为i,元素不超过j,个数为k的方案数。

具体递推式可参考代码。

[cpp] view plain copy
  1. #include<cstdio>  
  2. #include<cstring>  
  3. int n, k, s, ans;  
  4. int dp[156][21][11];  
  5. int main(){  
  6.     memset(dp,0,sizeof(dp));  
  7.     for(int i=1; i<=155; i++){  
  8.         for(int j=1; j<=20; j++){  
  9.             for(k=1; k<=10; k++){  
  10.                 if(k==1){  
  11.                     if(i<=j) dp[i][j][k] = 1;  
  12.                 }  
  13.                 else{  
  14.                     dp[i][j][k] = dp[i][j-1][k];  
  15.                     if(i>j)  dp[i][j][k] += dp[i-j][j-1][k-1];  
  16.                 }  
  17.             }  
  18.         }  
  19.     }  
  20.     while(~scanf("%d %d %d", &n, &k, &s)){  
  21.         if(!(n||k||s))  break;  
  22.         printf("%d\n", dp[s][n][k]);  
  23.     }  
  24.     return 0;  
  25. }  
0 0
原创粉丝点击