关闭

在n个数字中,找出所有和为SUM的组合

标签: integerstringimportclass算法
1708人阅读 评论(0) 收藏 举报
分类:
        题目具体是这样的,给定n个数字,比如{2,6,1,7,4,9},并给定一个SUM,比如SUM=20,在上述这6个数字中,挑出一些,使得他们的和等于SUM,把所有的组合都找出来。我们这个例子的结果就是:
1 + 2 + 4 + 6 + 7 = 20
1 + 4 + 6 + 9 = 20
4 + 7 + 9 = 20

具体代码如下:
import java.util.ArrayList;
import java.util.Arrays;

public class Main {
    
public static void main(String[] args) {
        
int[] a = {261749};
        
final int SUM = 20;
        ArrayList
<String> x = findExp(a, SUM);
        
for(String s:x) {
            System.out.println(s);
        }

    }


    
    
private static ArrayList<String> findExp(int[] a, final int SUM) {
        ArrayList
<String> result = new ArrayList<String>();    //存放的是所有找到的表达式
        Arrays.sort(a);    //对初始数据进行排序,便于算法的实现
        ArrayList<Integer> l = new ArrayList<Integer>();    //存放的是当前找到的合法的元素,这些元素的和小于SUM
        find(a, SUM, 0, l, result);    //通过递归的方式进行查找
        return result;
    }

    
    
    
//递归函数,每一次的动作很简单,
    
//在已经找到的n个元素的基础上,寻找第n+1个元素,
    private static void find(int[] a, final int SUM, int cur, ArrayList<Integer> l, ArrayList<String> result) {
        
int beg = l.size() == 0 ? 0 : l.get(l.size()-1)+1;    //当前元素的查找范围的起始位置
        for(int i=beg; i<a.length; i++{    //从起始位置到结束位置,查找合适的元素
            cur += a[i];    //在前面元素的和的基础上,加上当前元素
            if(cur<SUM) {    //如果仍然小于SUM,证明当前元素(第n+1个)合法,继续寻找第n+2个元素
                l.add(i);
                find(a, SUM, cur, l, result);
                cur 
-= a[i];    //消除第i个元素的影响,为了试验第i+1个元素做准备
                l.remove(l.size()-1);    //消除第i个元素的影响,为了试验第i+1个元素做准备
            }
else if(cur == SUM) {    //如果等于SUM,证明找到了表达式
                StringBuilder s = new StringBuilder();
                
for(int x:l) {    //构造表达式,并存入result
                    s.append(a[x]);
                    s.append(
" + ");
                }

                s.append(a[i]);
                s.append(
" = ");
                s.append(SUM);
                result.add(s.toString());
                
                
break;    //回溯到上一个状态
            }
else {    //如果大于SUM,也回溯到上一个状态
                break;
            }

        }

    }

}

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:871709次
    • 积分:5753
    • 等级:
    • 排名:第4468名
    • 原创:73篇
    • 转载:15篇
    • 译文:5篇
    • 评论:345条
    文章分类