算法训练 加法分解 Java题解 645 dfs

【问题描述】
　　给一个正整数n，输出它所有的正整数加法的分解方法。
　　注意：
　　1． 根据输入的要求决定交换加数的位置是否视为不同的分解方案。
　　2． 不分解也视为一种分解方案。
　　3． 按字典序输出所有分解方案，格式见样例。

【输入格式】
　　输入共1行，包含2个正整数n和m，之间用一个空格隔开。n表示待分解正整数，m是1或者2：
　　1表示交换加数的位置是否视为不同的分解方案；
　　2表示交换加数的位置是否视为相同的分解方案。
【输出格式】
　　输出若干行，每行表示一种分解方案。对于一种方案，先输出n，再输出一个“=”。然后输出分解的各数，不同的数之间用一个“+”连接。

样例输入

5 2

样例输出

5=1+1+1+1+1
5=1+1+1+2
5=1+1+3
5=1+2+2
5=1+4
5=2+3
5=5

【输入输出样例2】

样例输入

5 1

样例输出

5=1+1+1+1+1
5=1+1+1+2
5=1+1+2+1
5=1+1+3
5=1+2+1+1
5=1+2+2
5=1+3+1
5=1+4
5=2+1+1+1
5=2+1+2
5=2+2+1
5=2+3
5=3+1+1
5=3+2
5=4+1
5=5

数据规模和约定

　　对于50%的数据有M=1，另有50%的数据有M=2。对100%的数据，n≤15。

---

解题思路：

对于m=1的情况，相当于可以出现重复数字的“全排列”问题，只需要凑数到正好到n。

m=2时，相当与组合型的递归，要保证分解后的后一个不能小于前一个数，需要每次记录前一个数。

Java代码：

import java.io.*;
import java.util.*;

public class Main {
	static int n;
	static int sum;
	static ArrayList<Integer> list= new ArrayList<>();
	static StringBuilder ans = new StringBuilder();
	
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		String[] split = br.readLine().split(" ");
		n = Integer.parseInt(split[0]);
		int m = Integer.parseInt(split[1]);
		if(m == 1)
			dfs();
		else 
			dfs1(1);
		System.out.print(ans);
	}
	
	public static void dfs() {
		if(sum >= n) {
			if(sum > n)return;
			handle();
		}
		
		for(int i = 1; i <= n; i++) {
			list.add(i);
			sum += i;
			dfs();
			list.remove(list.size() - 1);
			sum -= i;
		}
	}
	
	public static void dfs1(int start) {
		if(sum >= n) {
			if(sum > n)return;
			handle();
		}
		
		for(int i = start; i <= n; i++) {
			list.add(i);
			sum += i;
			dfs1(i);
			list.remove(list.size() - 1);
			sum -= i;
		}
	}
	
	public static void handle() {
		ans.append(n + "=");
		for(int i = 0; i < list.size() - 1; i++) {
			ans.append(list.get(i) + "+");
		}
		ans.append(list.get(list.size() - 1) + "\n");
	}
}