蓝桥杯-第九届决赛——整理玩具

标题：整理玩具

小明有一套玩具，一共包含NxM个部件。这些部件摆放在一个包含NxM个小格子的玩具盒中，每个小格子中恰好摆放一个部件。  

每一个部件上标记有一个0~9的整数，有可能有多个部件标记相同的整数。  

小明对玩具的摆放有特殊的要求：标记相同整数的部件必须摆在一起，组成一个矩形形状。

如以下摆放是满足要求的：

00022
00033
44444  

12244
12244
12233

01234
56789

以下摆放不满足要求：

11122
11122
33311

111111
122221
122221
111111

11122
11113
33333

给出一种摆放方式，请你判断是否符合小明的要求。

输入
----
输入包含多组数据。  
第一行包含一个整数T，代表数据组数。 (1 <= T <= 10) 
以下包含T组数据。  
每组数据第一行包含两个整数N和M。  (1 <= N, M <= 10)  
以下包含N行M列的矩阵，代表摆放方式。  

输出
---
对于每组数据，输出YES或者NO代表是否符合小明的要求。  

【样例输入】
3  
3 5  
00022
00033
44444  
3 5  
11122
11122
33311
2 5  
01234
56789

【样例输出】
YES
NO
YES

资源约定：
峰值内存消耗（含虚拟机） < 256M
CPU消耗  < 1000ms

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

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

 思路：把每组中的数按顺序收集，然后再一个一个判断是否符合。既然是矩形，就保存它开始的坐标和结束的坐标。

import java.util.HashSet;
import java.util.Iterator;
import java.util.Scanner;
import java.util.Set;

public class Main {

public static void main(String[] args) {
		
		Scanner in = new Scanner(System.in);
		int t = in.nextInt();
		int [] fal = new int [t];//保存输出的结果 1YES 0 NO
		for(int i=0;i<t;i++){
			int n = in.nextInt();
			int m = in.nextInt();
			int [][] a = new int [n][m];
			Set<Integer> set = new HashSet<Integer>();//计算一下这组数据中有几个数
			for(int j=0;j<n;j++){
				String b = in.next();
				for(int k=0;k<m;k++){
					a[j][k] = b.charAt(k)-48;
					set.add(a[j][k]);
				}
			}
			if(show(a,set)){//图、数
				fal[i]=1;
			}else{
				fal[i]=0;
			}
			set.clear();
		}
		for(int i=0;i<t;i++){//输出结果
			if(fal[i]==1){
				System.out.println("YES");
			}else{
				System.out.println("NO");
			}
		}
	}
	/**
	 * 
	 * @param a 图
	 * @param set 这组数据有几个
	 * @return
	 */
	private static boolean show(int [][] a,Set<Integer> set) {

		boolean fa = true;
		Iterator<Integer> it = set.iterator();
		while(it.hasNext()){//0 2 3 4
			int x1 = 0;//初始行
			int x2 = 0;//结束行
			int y1 = 0;//初始列
			int y2 = 0;//结束列
			int t = it.next();
			int fal = 0;
			int c2 = 0;
			for(int i=0;i<a.length;i++){
				for(int j=y1;j<a[0].length;j++){
					if(fal!=0){
						if(a[i][j]==t&&fal==1&&j!=a[0].length-1){//列结束的位置
							y2 = j;//2
//							break;
						}else if(j==a[0].length-1){
							if(a[i][j]==t&&fal==1){
								y2 = j;
							}
							fal = 2;
						}else if(a[i][y1]==t&&fal==2){
//							fal=2;
							c2++;
							x2=x1+c2;
							break;
						}
					}
					if(fal==0){
						if(a[i][j]==t){//开始的位置
							fal = 1;
							x1 = i;//0//开始的位置行
							y1 = j;//0//开始的位置列
							x2 = x1;
							y2 = y1;
						}
					}
				}
			}
			if(!pd(a,x1,y1,x2,y2,t)){
				fa = false;
			}
		}
		return fa;
	}
	/**
	 * 
	 * @param a 二位数组
	 * @param x1 起始行
	 * @param y1 起始列
	 * @param x2 结束行
	 * @param y2 结束列
	 * @param b 哪个玩具
	 * @return
	 */
	public static boolean pd(int [][] a,int x1,int y1,int x2,int y2,int b){
//		System.out.println("x1:"+x1+" y1:"+y1+" x2:"+x2+" y2:"+y2+" t:"+b);
		for(int i=0;i<a.length;i++){
			for(int j=y1;j<a[0].length;j++){
				if(i<x1||i>x2||j<y1||j>y2){
					if(a[i][j]==b){
						return false;
					}
				}else{
					if(a[i][j]!=b){
						return false;
					}
				}
			}
		}
		return true;
	}
}