和尚挑水问题

前几天做了次笔试题，里面有条和尚挑水的问题，感觉对DFS的理解很有帮助。

题目：

某寺庙里7个和尚:轮流挑水，为了和其他任务不能冲突，例如：各人将有空天数列出如下表
和尚1: 星期二,四;
和尚2: 星期一,六;
和尚3: 星期三,日;
和尚4: 星期五;
和尚5: 星期一,四,六;
和尚6: 星期二,五;
和尚7: 星期三,六,日;

此时的输入就是 

0 1 0 1 0 0 0

1 0 0 0 0 1 0

0 0 1 0 0 0 1

0 0 0 0 1 0 0

1 0 0 1 0 1 0

0 1 0 0 1 0 0

0 0 1 0 0 1 0

请将所有合理的挑水时间安排表 

思路：利用DFS将每种方案都列出来，如果安排到某一天的时候发现没有和尚有空，就立刻返回寻找其他方案。

代码如下：

import java.util.*;

public class Main {

 

publicstatic void main(String[] args) {

//TODO Auto-generated method stub

Scannersc = new Scanner(System.in);

int[][]in = new int[7][7];

for(inti = 0;i < 7;i++){

for(intj = 0;j < 7;j++){

in[i][j]= sc.nextInt();

}

}

 

ArrayList<ArrayList<Integer>>res = new ArrayList<ArrayList<Integer>>();

ArrayList<Integer>occupy = new ArrayList<Integer>();

 

dfs(res,in,occupy,0);

System.out.println(res.size());

for(inti = 0;i < res.size();i++){

System.out.println((res.get(i).get(0)+1)+""+(res.get(i).get(1)+1)+" "+

(res.get(i).get(2)+1)+""+(res.get(i).get(3)+1)+" "+

(res.get(i).get(4)+1)+""+(res.get(i).get(5)+1)+" "+

(res.get(i).get(6)+1));

}

//return;

 

 

}

publicstatic void dfs(ArrayList<ArrayList<Integer>> res,int[][]in,ArrayList<Integer> occupy,int day){

System.out.println("day is "+day);

 

if(day==6){

for(inti = 0;i < 7;i++){

if(in[i][day]==1){

if(!occupy.contains(i)){

occupy.add(i);

ArrayList<Integer>tmp = new ArrayList<Integer>(occupy);

//tmp.add(i);

res.add(tmp);

//System.out.println(res.size());

//dfs(res,in,occupy,day+1);

occupy.remove(occupy.size()-1);//一定要记得去掉，不然只会出一个正确的可能

 

return;

}

}

}

return;

}else{

for(inti = 0;i < 7;i++){

if(in[i][day]==1){

if(!occupy.contains(i)){

//System.out.println("iis "+i+" day is "+day);

occupy.add(i);

dfs(res,in,occupy,day+1);

occupy.remove(occupy.size()-1);

}//如果不满足，千万不要return

}

}

return;

}

}

 

}