题目及翻译
题面
LZY今天又突发奇想了,他想随意给出两个数字n和k,请你找出所有N个数字相加和为K的组合数量。组合中只含有1——9的数字,并且不允许重复。
输入
测试样例由多组测试数据组成。每组测试样例第一行输入两个正整数n ( 1 <= n <= 9 ) 和 k < ( 1 <= k <= 45 )。
输出
输出所有符合题目要求的组合数量。如果组合不存在,请输出 No list
输入样例
3 7
3 9
输出样例
1
3
题目思路
给学弟学妹写篇题解,小不点们加油!!
有不懂的地方欢迎留言或者微信找我询问~
题目要求里已经写清楚退出条件了,刚好用 n 个数字 且 总和为 k
那么我们只需要记录三个值,一个是当前值,一个是当前总和,一个是当前数字数量
有几个剪枝优化:
1:如果当前没有到达正确答案,那么 步数 大于等于 n 都意味着之后没可能了
2:如果当前没有到达正确答案,那么 总和 大于等于 k 都意味着之后没可能了
3:从当前值 + 1 开始从小到大搜索,这样就不用再标记是否有数字被用过,因为逼自己小的数字都已经在之前的计算中算完了
注意事项
没有可能时,要输出No list 我没看仔细题目就栽在这了
AC代码
C/C++(几乎没有代码变更)
用时0MS 内存1120K 长度399B
#include<stdio.h>
int n,k,res;//n和k是退出条件,res是答案
int dfs(int cur,int cnt,int step){//cur是当前位置,cnt是当前总和,step是当前所用数字个数
if(step == n && cnt == k)return 1;//如果刚好数字用了n个且总和到了k就返回1,代表这个答案被记录过
if(step >= n)return 0;//如果步数大于等于n还没找到正确答案,说明这之后没可能了
if(cnt >= k)return 0;//如果用了大于等于k的数还没找到正确答案,说明这之后没可能了
int sum = 0;//用于记录答案
for(int i=cur+1;i<=9;++i){//遍历从当前位置之后的数字开始,一直到9的所有数字
sum += dfs(i,cnt + i,step + 1);//统计每个情况的值
}
return sum;//返回当前位置的所有可能
}
int main(){
while(~scanf("%d %d",&n,&k)){
res = dfs(0,0,0);//一切从0开始!!cooooooooooooooooooooooool~
if(res)printf("%d\n",res);//如果有答案就输出答案
else printf("No list\n");//如果没答案就输出No list
}
return 0;
}
本文作者 优快云@扶她小藜
个人主页链接 https://blog.youkuaiyun.com/weixin_44579869