POJ - 多少种取法(递归)

280 阅读1分钟

描述

给定三个正整数 m, n, s 问从 1 到 m 这 m 个数里面取 n 个不同的数,使它们和是 s , 有多少种取法

输入

多组数据

输入的第一行是整数 t ,表示有 t 组数据

此后有 t 行,每行是一组数据

每组数据就是三个正整数,m, n, s ( n <= 10, s <= 20)

输出

对每组数据,输出答案

样例输入

5
13 4 20
12 5 18
1 1 1
1 2 1
119 3 20

样例输出

22
3
1
0
24

提示

用函数ways(m,n,s)表示 从1到m这m个数里面取n个不同的数,使它们和是s的取法总数
显然,必须取m个数,不能不取(除非m == 0)

  1. 考虑如果 m > s, 问题可以等价于什么?
  2. 对于m<= s的情况,把所有的取法分成两类:
    第一类: 取m。则取m后,剩下的问题变成什么?
    第二类: 不取m,那么剩下的问题变成什么?
  3. 注意边界条件(即递归终止条件,即不需要递归的条件)
    边界条件一般是 n,m,s = 0, = 1 之类的情况。

例如:从 1-m这m个数里面,取0个数,使得它们的和是0,有几种取法? 答案是1。
从 1到m这m个数里面,取0个数,使得它们的和是s(s>0),有几种取法? 答案是0。无解对应的答案就是0.
当 m < n时,答案是0,因为没法取n个数
当 m = 0时,只要m和s有一个不是0,ways(m,n,s)就应该返回0。

递归的时候,函数的参数会减少,如果会出现某个参数一直没完没了减少下去,那就不对了。因此,边界条件一定要考虑周全,确保递归可以终止。

边界条件可以有多种写法。

题解

def solve(m, n, s):
    if m == 0:
        if n == 0:
            if s == 0:
                return 1
            else:
                return 0
        else:
            return 0

    if m > s:
        return solve(s, n, s)
    else:
        return solve(m - 1, n - 1, s - m) + solve(m - 1, n, s)


t = int(input())
for i in range(t):
    l = input().split()
    m, n, s = int(l[0]), int(l[1]), int(l[2])
    print(solve(m, n, s))

来源: OpenJudge - 030:多少种取法