Problem B: 组数总和

Description

LZY今天又突发奇想了，他想随意给出两个数字n和k，请你找出所有N个数字相加和为K的组合数量。组合中只含有1——9的数字，并且不允许重复。

Input

测试样例由多组测试数据组成。每组测试样例第一行输入两个正整数n ( 1 <= n <= 9 ） 和 k < ( 1 <= k <= 45 ）。

Output

输出所有符合题目要求的组合数量。如果组合不存在，请输出 No list

Sample Input

3 7
3 9

Sample Output

1
3
代码

#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
using namespace std;
int k,n; //取 k 个数 和为 n 
int vis[10]; 
int flag = 0; 
int ans = 0;//总和 
void dfs(vector<int>& temp,int sum){    
 if(temp.size() == k){      
  if(sum == n){           
   flag = 1;            
   ans ++;        
  }      
  return ;
 }  
 for(int i = 1 ; i <= 9 ; i ++){      
  if(vis[i] == 1)
   continue;     
  if(temp.size() > 0 && i < temp[temp.size() - 1])          
   continue;     
  if(sum + i > n)           
   continue;      
  vis[i] = 1;       
  sum += i;        
  temp.push_back(i);      
  dfs(temp , sum);        
  temp.pop_back();      
  sum -= i;        
  vis[i] = 0;    
 }
}

int main(){    
 while(cin >> k >> n){
  memset(vis,0, sizeof(vis));   
  flag = 0;      
  ans = 0;      
  vector<int> temp;      
  dfs(temp,0);       
  if(!flag)        
   cout << "No list" << endl;      
  else           
   cout << ans << endl;    
 }
 return 0;
}

思路

temp.pop_back(); sum -= i; vis[i] = 0; //回溯
if(vis[i]==1) continue; //判断是否被使用过
if(temp.size()k){ //递归终止条件
if(sumn){ //判断是否满足相加和
flag = 1;
ans ++;
}
return ;
}
for(int i = 1 ; i <= 9 ; i ++){
if(vis[i] == 1) //判断是否使用过这个数字
continue;
if(temp.size() > 0 && i < temp[temp.size() - 1])
continue;
if(sum + i > n)
continue;