题目描述
输入两个整数 n 和 m,从数列1,2,3.......n 中随意取几个数,使其和等于 m ,要求将其中所有的可能组合列出来
输入描述:
每个测试输入包含2个整数,n和m
输出描述:
输入
1 4
2 3
很想吐槽这种需要判断行尾空格问题的题目,贼麻烦。
测试只A了70%,其他测试数据由于超时W。
输入两个整数 n 和 m,从数列1,2,3.......n 中随意取几个数,使其和等于 m ,要求将其中所有的可能组合列出来
输入描述:
每个测试输入包含2个整数,n和m
输出描述:
按每个组合的字典序排列输出,每行输出一种组合
输入
5 5
1 4
2 3
5
import java.util.*;
public class Main{
public static void main(String [] args){
Scanner in = new Scanner(System.in);
int n = in.nextInt();
int m = in.nextInt();
int [] visited = new int [n + 1];
getRes(n, m, 1, visited);
}
public static void getRes(int n, int m, int index, int [] visited){
if(m == 0){
printRes(n, index, visited);
return;
}
for(int i = index; i <= n; ++i){
if(m >= i){
visited[i] = 1;
getRes(n, m - i, i + 1, visited);
visited[i] = 0;
}else{
if(m != 0)
return;
}
}
}
public static void printRes(int n, int index, int [] visited){
for(int k = 1; k <= n; ++k){
if(visited[k] == 1){
if(k == index - 1)
System.out.print(k);
else
System.out.print(k + " ");
}
}
System.out.println();
}
}
很想吐槽这种需要判断行尾空格问题的题目,贼麻烦。
类似的这种组合问题都可以使用这样的DFS思想。
还有这样一道同类型的题目:
题目描述
给你六种面额1、5、10、20、50、100元的纸币,假设每种币值的数量都足够多,编写程序求组成N员(N为0-10000的非负整数)的不同组合的个数。
输入描述:
输入为一个数字N,即需要拼凑的面额
输出描述:
输出也是一个数字,为组成N的组合个数。
示例1
输入
5
输出
2
也可以这样去思考,但是题设要求N的范围为0-10000,因为这道题目属于可重复型选择元素进行装填,所以10000意味着10000层递归栈,时间复杂度跟空间复杂度都会爆炸,对于这种可重复性添加的问题,可以考虑使用背包问题。
以下为使用DFS递归的代码:
import java.util.*;
public class Main {
public static int res = 0;
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int N = in.nextInt();
int [] money = {1,5,10,20,50,100};
getRes(N, money, 0);
System.out.println(res);
}
public static void getRes(int n, int [] money, int index){
if(n == 0){
res++;
return;
}else if(n < 0){
return;
}
for(int i = index; i <= 5; ++i){
if(n - money[i] < 0) return;
getRes(n-money[i], money, i);
}
}
}
测试只A了70%,其他测试数据由于超时W。
使用背包算法就会AC,但是效率也高不到哪里去,还是香港记者效率高。