蓝桥杯算法训练 数字游戏 Java DFS

资源限制
时间限制：1.0s 内存限制：256.0MB
问题描述
　　给定一个1～N的排列a[i]，每次将相邻两个数相加，得到新序列，再对新序列重复这样的操作，显然每次得到的序列都比上一次的序列长度少1，最终只剩一个数字。
　　例如:
　　3 1 2 4
　　4 3 6
　　7 9
　　16
　　现在如果知道N和最后得到的数字sum，请求出最初序列a[i]，为1～N的一个排列。若有多种答案，则输出字典序最小的那一个。数据保证有解。
输入格式
　　第1行为两个正整数n，sum
输出格式
　　一个1～N的一个排列
样例输入
4 16
样例输出
3 1 2 4
数据规模和约定
　　0<n<=10

import java.util.Scanner;
 /*
  * 
  * */
public class Main {
	
	static int n;//初始的n个数
	static int sum;//相加的结果
	static int arr1[];
	static boolean bool = true;//标记是否找到n个元素
	
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		n = sc.nextInt();
		sum = sc.nextInt();
		
		int array[] = new int[n];
		int visit[] = new int[n+1]; //访问标记
		
		dfs(0, array, visit);
	}
	
	//step:当前已经遍历的元素数；array：存放当前遍历的元素； visit：标记当前哪些元素已被访问； 
	public static void dfs(int step,int arr[], int vis[]){
		if(step == n){//1.找到n个数字，检查这n个数相加是否等于sum
			int arr1[] = new int[n];
			for(int i=0;i<n;i++){//先将array数组中的元素 放到arr1
				arr1[i] = arr[i];
			}//for
			
			for(int i=1;i<n;i++){//累加到arr1[0]
				for(int j=0;j<n-i;j++){
					arr1[j]=arr1[j]+arr1[j+1];
				}
			}//for
			
			if(arr1[0] == sum){//累加后与sum相等,则输出
				for(int i:arr){
					System.out.print(i+" ");
				}
				bool = false;//已经找到符合要求的n个元素
				return;
			}else{//累加后与sum不相等，退出重新查找
				return;
			}
			
		}//if
		if(bool == true){//2.没有找到符合要求的n个元素
			for(int i=1;i<=n;i++){
				if(vis[i]==0){//i值没有使用过
					arr[step]=i;//存储当前访问的元素
					vis[i]=1;
					dfs(step+1,arr,vis);//从当前这个元素起，重新进行遍历
					vis[i]=0;
				}
			}
		}
		return;
	}
}