DFS解决组合题目

9 篇文章 0 订阅
题目描述
输入两个整数 n 和 m,从数列1,2,3.......n 中随意取几个数,使其和等于 m ,要求将其中所有的可能组合列出来
输入描述:
每个测试输入包含2个整数,n和m
输出描述:

按每个组合的字典序排列输出,每行输出一种组合


示例1
输入

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,但是效率也高不到哪里去,还是香港记者效率高。


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值