第七届蓝桥杯决赛JavaA组第六题_生成树计数

生成树计数
 
给定一个 n*m 的格点图，包含 n 行 m 列共 n*m 个顶点，相邻的顶点之间有一条边。
【图1.png】给出了一个3*4的格点图的例子。

如果在图中删除部分顶点和其相邻的边，如上图删除第2行第3列和第3行第1列的顶点后，如【图2.png】所示。

图的生成树指包含图中的所有顶点和其中的一部分边，使得任意两个顶点之间都有由边构成的唯一路径。如果两个生成树包含有不同的边即被认为不同，则上图中共有31种不同的生成树，其中a边不选有10种，a边选有21种。
给出格点图中保留的顶点的信息，请计算该图一共有多少种不同的生成树。

【输入格式】
输入的第一行包含两个整数n, m，用空格分隔，表示格点图的行数和列数。
接下来n行，每行m个字母（中间没有分隔字符），每个字母必然是大写E或大写N，E表示对应的顶点存在，N表示对应的顶点不存在。保证存在至少一个顶点。

【输出格式】
输出一行，包含一个整数，表示生成树的个数。答案可能很大，你只需要计算答案除以1000000007的余数即可。

【样例输入】
 3 4
EEEE
EENE
NEEE

【样例输出】
31

【数据规模与约定】
对于10%的数据，1<=n<=2。
对于30%的数据，1<=n<=3。
对于40%的数据，1<=n<=4。
对于50%的数据，1<=n<=5。
另有20%的数据，1<=n*m<=12。
另有10%的数据，1<=m<=15。
对于100%的数据，1<=n<=6，1<=m<=100000。

资源约定：
峰值内存消耗 < 256M
CPU消耗  < 8000ms

请严格按要求输出，不要画蛇添足地打印类似：“请您输入...” 的多余内容。

所有代码放在同一个源文件中，调试通过后，拷贝提交该源码。
注意：不要使用package语句。不要使用jdk1.7及以上版本的特性。
注意：主类的名字必须是：Main，否则按无效代码处理。

import java.util.HashSet;
import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;
import java.util.Set;

public class Main {

	public static int all_point_num=0;             ///点的个数
	public static Set<String>set=new HashSet<>();  ///去除重复的元素        
	
	
	public static void main(String[] args) {
		Scanner scanner=new Scanner(System.in);
		int m=scanner.nextInt();
		int n=scanner.nextInt();
		char [][]matrix_point=new char[m][n];
		boolean [][]matrix_edge=new boolean[m*2][n];
		for(int i=0;i<m;i++){
			String temp=scanner.next();
			for(int j=0;j<n;j++){
				matrix_point[i][j]=temp.charAt(j);
				if(matrix_point[i][j]=='E')all_point_num++;
			}
		}
		set_matrix_enge(matrix_point, matrix_edge);
		can_build_tree(matrix_edge, matrix_point);
		System.out.println(set.size()%1000000007);
	}
	
	public static void set_matrix_enge(char[][]matrix_point,boolean[][] matrix_edge){
		///我们只检查每个节点的右侧和下侧，如果右侧和下侧是'E'，为其填上边
		for(int i=0;i<matrix_point.length;i++){
			for(int j=0;j<matrix_point[i].length;j++){
				if(matrix_point[i][j]=='E'){
					if(j+1<matrix_point[i].length&&matrix_point[i][j+1]=='E')matrix_edge[i*2][j]=true;
					if(i+1<matrix_point.length&&matrix_point[i+1][j]=='E')matrix_edge[i*2+1][j]=true;
				}
			}
		}
	}
	
	public static void can_build_tree(boolean[][]matrix_edge,char[][]matrix_point){
		for(int i=0;i<matrix_edge.length;i++){
			for(int j=0;j<matrix_edge[i].length;j++){
				if(matrix_edge[i][j]){            ///找到一条边，将这条边删除，并检查是否是生成树
					boolean[][]current_matrix_edge=copy_matrix_edge(matrix_edge);
					current_matrix_edge[i][j]=false;
				    if(is_a_tree_bfs(current_matrix_edge, matrix_point)){
				    	String temp_string=matrix_edge_to_string(current_matrix_edge);
				    	if(temp_string!="")
				    	set.add(temp_string);
				    	can_build_tree(current_matrix_edge, matrix_point);
				    }
					
				}
			}
		}
	}

	
	public static String matrix_edge_to_string(boolean[][]matrix_edge){
		String result="";
		int temp_sum=0;
		for(int i=0;i<matrix_edge.length;i++){
			for(int j=0;j<matrix_edge[i].length;j++){
				if(matrix_edge[i][j]==true){
					result+='1';
					temp_sum++;
				}else{
					result+='0';
				}
			}
		}
		if(temp_sum!=all_point_num-1){
			return "";
		}
		return result;
	}
	

	public static boolean is_a_tree_bfs(boolean[][]current_matrix_edge,char[][]matrix_point){
		boolean result=false;
		int current_point_num=0;
		char[][]matrix_mark=new char[matrix_point.length][matrix_point[0].length];
		Queue<Integer>queue=new LinkedList<Integer>();
		for(int i=0;i<matrix_point.length;i++){
			for(int j=0;j<matrix_mark[i].length;j++){
				if(matrix_point[i][j]=='E'){              ///找到第一个节点，将其放入到
					queue.add(i);
					queue.add(j);
					current_point_num++;
					matrix_mark[i][j]='E';
					while(!queue.isEmpty()){
						int current_i=queue.remove();
						int current_j=queue.remove();
						if(current_j+1<matrix_point[current_i].length&¤t_matrix_edge[current_i*2][current_j]==true&&matrix_point[current_i][current_j+1]=='E'&&matrix_mark[current_i][current_j+1]!='E'){         ///右侧有节点  右侧的路是通的，并且右侧是节点  从未走过  
							queue.add(current_i);
							queue.add(current_j+1);
							current_point_num++;
							matrix_mark[current_i][current_j+1]='E';
							//System.out.println("当前的坐标是："+current_i+","+current_j+"  他的右侧是可行的");
						}
						if(current_i+1<matrix_point.length&¤t_matrix_edge[current_i*2+1][current_j]==true&&matrix_point[current_i+1][current_j]=='E'&&matrix_mark[current_i+1][current_j]!='E'){          ///下侧有节点   下侧是通的  下侧是节点  从未走过     
							queue.add(current_i+1);
							queue.add(current_j);
							current_point_num++;
							matrix_mark[current_i+1][current_j]='E';
							//System.out.println("当前的坐标是："+current_i+","+current_j+"  他的下侧是可行的 ");
						}
						if(current_j-1>=0&¤t_matrix_edge[current_i*2][current_j-1]==true&&matrix_point[current_i][current_j-1]=='E'&&matrix_mark[current_i][current_j-1]!='E'){         ///左  
							queue.add(current_i);
							queue.add(current_j-1);
							current_point_num++;
							matrix_mark[current_i][current_j-1]='E';
							//System.out.println("当前的坐标是："+current_i+","+current_j+"  他的左侧是可行的");
						}
						if(current_i-1>=0&¤t_matrix_edge[current_i*2-1][current_j]==true&&matrix_point[current_i-1][current_j]=='E'&&matrix_mark[current_i-1][current_j]!='E'){          ///上     
							queue.add(current_i-1);
							queue.add(current_j);
							current_point_num++;
							matrix_mark[current_i-1][current_j]='E';
							//System.out.println("当前的坐标是："+current_i+","+current_j+"  他的上侧是可行的 ");
						}
					}
					if(current_point_num==all_point_num)
						return true;
					else
						return false;
				}
			}
		}
		return result;
	}
	
	public static boolean[][]copy_matrix_edge(boolean[][]matrix_edge){
		boolean [][]result=new boolean[matrix_edge.length][matrix_edge[0].length];
		for(int i=0;i<matrix_edge.length;i++){
			for(int j=0;j<matrix_edge[i].length;j++){
				result[i][j]=matrix_edge[i][j];
			}
		}
		return result;
	}

}