牛牛和牛妹在感叹现代科技日新月异的同时,也不忘关注着环保事业。 今天,牛牛和牛妹一同来到了戈壁滩种树,他们决定在同一排种下一共 n 棵小树,已知每棵小树的存活率相同,现在,牛牛和牛妹想要知道,在这 n 棵树中,恰好有 k 棵树存活时,最终这排小树形成的风景线会有多少种不同的情况。 由于最终答案可能很多,所以只需要输出对 10 ^ 9+ 7 取模后的结果。
本题为多组测试数据,第一行输入一个正整数 T( 1≤ T≤ 1000) ,代表测试数据组数。 对于每组测试数据,一行输入两个正整数 n, k( 1≤ k≤ n≤ 1000) ,分别代表种下的树的数量以及最终恰好有 k 棵树存活。
对于每组测试数据,一行输出一个整数代表一共多少种情况。
1 2 1
2
考点:组合数学
数据规模 n ≤ 1e3 | 限制 1 秒 / 64MB | 标准输入输出
推荐方向:组合数学
本题切入点
n 棵里恰好 k 棵存活,方案数就是组合数 C(n,k),预处理阶乘与逆元后取模输出。
用计数原理(加法/乘法原理)与组合数公式统计方案数。
思路框架(组合数学 通法 · 非本题专属)
实现要点:阶乘预处理 + 费马小定理求逆元可以在 O(1) 内算任意组合数。
复杂度:时间 O(n) 预处理 / O(1) 查询 | 空间 O(n)
该范式的通法易错点
对照本题
样例 1:输入 1 / 2 1 → 输出 2
如果存活的是第一棵小树,那么最终这两棵小树形成的风景线是 茂盛的小树戈壁滩(小树已死亡)
如果存活的是第二棵小树,那么最终这两棵小树形成的风景线是 戈壁滩(小树已死亡)茂盛的小树
显然,种了两棵树存活一棵,只有上述两种情况。
解析由校招宝本地引擎整理(依据源站考点标签 / 人工判题标注 / 题面规模信号),非官方题解,仅供思路参考。
本题来源:贝壳找房2023届校招算法卷3。